liuwei16 / ALFNet

Code for 'Learning Efficient Single-stage Pedestrian Detectors by Asymptotic Localization Fitting' in ECCV2018
129 stars 75 forks source link

Setting batch size to 10 #22

Open Kyanite123 opened 4 years ago

Kyanite123 commented 4 years ago

Hi Liu,

Thank you for sharing with the community your excellent work!

I'm trying to reconstruct the mobilenet model, and using the default crop size of 640x1280 px. I have only 1 GPU with memory of 11.7G (same model as in the paper GTX 1080 Ti).

It seems that this memory size allows me a maximum batch size of 4. When I try to increase to 10 I get out of memory errors. How were you able to increase your batch size to 10 ?

Many thanks!!