ajbrock / BigGAN-PyTorch

The author's officially unofficial PyTorch BigGAN implementation.
MIT License
2.84k stars 470 forks source link

sampling out of memory #87

Open jiyuwangbupt opened 3 years ago

jiyuwangbupt commented 3 years ago

When I executed the sample_BigGAN_bs256x8.sh script, an ”out of memory“ error occurred. However, the weights used did not report "out of memory“errors during training. image image image image

I use 4 p40 GPUs for training and sampling. Each GPU memory is about 22G.

ElaineGxy commented 2 years ago

I have the same question,emmmmm........