-
Notifications
You must be signed in to change notification settings - Fork 1
About training speed #4
Copy link
Copy link
Open
Description
Hello! Thanks for your excellent work.
I used the training command you provided (python train_dsr.py --gpu_id 0 --obj_id $i --lr 0.0002 --bs 16 --epochs 120 --data_path $MVTEC3D_PATH --out_path $OUT_PATH --run_name $RUN_NAME). It took around 14 hours to complete the training process for each single category. I trained using an RTX 3090 (24GB). Is this speed considered normal?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels