GroupFacePytorch icon indicating copy to clipboard operation
GroupFacePytorch copied to clipboard

Published 20 hours ago • verified publisher icon leoluopy

how to support higher batch size while training?

Open rlrahulkanojia opened this issue 4 years ago • 2 comments

Tested with decreasing image size, but still not enough.

rlrahulkanojia avatar Jan 19 '21 08:01 rlrahulkanojia

in this repository , 224x224 is preferred

leoluopy avatar Jan 22 '21 01:01 leoluopy

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?

rlrahulkanojia avatar Jan 22 '21 06:01 rlrahulkanojia