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