search

andreas128 / RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022
1.88k stars 153 forks source link

How much memory do i need to run the example? #25

Open yycloudywind opened 1 year ago

yycloudywind commented 1 year ago

My GPU is GTX1650 with 4GB of memory, and every time I run the example, I run out of memory. So is there a way to reduce the memory burden in the yaml parameters that I'm running, or do I need to upgrade my GPU to support the running conditions?

Zanghu-ze commented 5 months ago

hi yy, I think you should change the card. When I run the test.py on my A6000 card. The nvtop shows that it costs about 7GB memory. Hoping this helps you!