About the gpu memory usage of the training process

[zy-charon]

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]

hi bro, have you figured out the problem?