search

openai / guided-diffusion

MIT License
6.03k stars 803 forks source link

About the gpu memory usage of the training process #86

Open zy-charon opened 1 year ago

zy-charon commented 1 year ago

Hello, I have reproduced the experiment of the DDPM basic model before. When I use a picture of 128*128 size and batchsize is set to 2, the gpu memory usage is about 18G; when I use your code, the same setting, gpu The usage is about 5G, is this normal, or maybe there is something wrong with my settings?

chensming commented 1 year ago

hi bro, have you figured out the problem?