Setting batch size to 10

[Kyanite123]

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!!