Limit to GPU memory

[DoubleKing0625]

Hi,

I run your training code for DeepFashion.
And OOM error happens when i set batch size as 6 for MODEL101_DF_Encoder_GAN_BodyROI7.
Here i only can set batch size as 1, and i will take almost 8.5G memory per GPU.
So how to use 6 batch size for training?

Thanks in advance