denoising-diffusion-pytorch icon indicating copy to clipboard operation
denoising-diffusion-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lucidrains

How much GPU memory is needed to run this code？

Open Echo-jyt opened this issue 1 year ago • 2 comments

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

Echo-jyt avatar Mar 04 '23 06:03 Echo-jyt

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

lhaippp avatar Mar 07 '23 01:03 lhaippp

Thank you very much！

Echo-jyt avatar Mar 09 '23 08:03 Echo-jyt