wyhuai / DDNM

[ICLR 2023 Oral] Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

What configuration is required for model training?

qiongshuai opened this issue · comments

Your work is great! What kind of GPU is required for model training? Can your model be trained on 4 12GB GPUs?

Actually, we use pre-trained models from others. Of course, you can train your own model, I think 4 12GB GPUs are ok.