how to support higher batch size while training?

leoluopy / GroupFacePytorch

none official pytorch implement of CVPR2020 GroupFace

24 stars 1 forks source link

how to support higher batch size while training? #3

Open rlrahulkanojia opened 3 years ago

rlrahulkanojia commented 3 years ago

Tested with decreasing image size, but still not enough.

leoluopy commented 3 years ago

in this repository , 224x224 is preferred

rlrahulkanojia commented 3 years ago

I changed the dimensions to 112*112 in the network but still the maximum batch size i am able to get is 16 on 12GB GPU. I meant to ask how can I increase batch size to say 200 on same GPU?