Out of memory at start - Githubissues

lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two

MIT License

1.62k stars 220 forks source link

Out of memory at start #98

Open WhereAmO opened 3 years ago

WhereAmO commented 3 years ago

No matter what as soon as I start training it stops and says it's out of memory after about 5 seconds, I can successfully run the stylegan version of this (at least until the 23rd epoch and then it throws another out of memory error, even with the --no-pl-reg flag.) so I find that a bit odd. Any way to fix it?

yilmazgencc commented 2 years ago

Hello, I solved the problem by adding "--num-workers 0". I hope it helps you.