Where to set training batch size
GuanqiHe opened this issue · comments
When I run python train.py --task tracking
, the program raises cuda out of memory. I would like to ask where to set the training batch size to reduce the cuda memory requirement?
Please check the config
inside the training script.