MobileStyleGAN.pytorch icon indicating copy to clipboard operation
MobileStyleGAN.pytorch copied to clipboard

About batch_size during training

Open zhangziwenHIT opened this issue 2 years ago • 0 comments

You mentioned in the paper that setting batch_size==8 on four 2080Ti takes about three days to train. image But when I set batch_size>=4 on V100, I get an error. image The problem is the same as in this issue.https://github.com/bes-dev/MobileStyleGAN.pytorch/issues/23 image Is there something wrong with my configuration?As far as I know, the 2080Ti has 11GB of memory, while the V100 I used has 32GB. This is puzzling to me.Looking forward to getting your answer, thank you!

zhangziwenHIT avatar Jan 10 '24 08:01 zhangziwenHIT