Huge memory for wgan - Githubissues

yyqqss09 / ldct_denoising

MIT License

28 stars 6 forks source link

Huge memory for wgan #6

Open mohaEs opened 4 years ago

mohaEs commented 4 years ago

I have a GPU with a 32GB memory ram. I can not train the WGAN even with the batchsize=1. OOM error will occur. I was wondering which module needs the most memory and is there any modification for making the network simpler?

yyqqss09 commented 4 years ago

32Gb should be large enough. When you do the training, do you do it on whole image or small image patches? What is the size of image patches?