Disentangled-Person-Image-Generation icon indicating copy to clipboard operation
Disentangled-Person-Image-Generation copied to clipboard

Published 20 hours ago • verified publisher icon charliememory

Limit to GPU memory

Open DoubleKing0625 opened this issue 6 years ago • 0 comments

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

DoubleKing0625 avatar Jun 07 '18 04:06 DoubleKing0625