How much GPU memory is needed to run this code？

lucidrains / denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

MIT License

8.36k stars 1.03k forks source link

Open Echo-jyt opened 1 year ago

Echo-jyt commented 1 year ago

I am a newbie. I want to konw how much GPU memory is needed to run this code？

lhaippp commented 1 year ago

It mainly depends on your batch size and input resolution, such as

image_size=128,  train_batch_size=14, it takes about 11G GPU memory

Echo-jyt commented 1 year ago

Thank you very much！