ajbrock / BigGAN-PyTorch

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

Lower memory requirements on data prep #101

Open gcervantes8 opened 1 year ago

gcervantes8 commented 1 year ago

I kept getting an Out of Memory Error (RAM) when preparing the dataset on ImageNet 128x128 with 32 GB of RAM. Since the Pool and Labels are currently unused, I think it's worth commenting them out for now until they start being used again.