Fayeben / GenerativeDiffusionPrior

Generative Diffusion Prior for Unified Image Restoration and Enhancement (CVPR2023)
Apache License 2.0
276 stars 30 forks source link

I would like to ask what kind of graphics card is recommended for training such a model. #3

Closed owl-of-pastos closed 1 year ago

owl-of-pastos commented 1 year ago

Is 3090 enough?

Fayeben commented 1 year ago

Hi, the model we used is provided by OpenAI/Guided_diffusion. It does not need to be trained.

MrWan001 commented 1 year ago

Hi, the model we used is provided by OpenAI/Guided_diffusion. It does not need to be trained.

Hello!OpenAI / Guided_ Diffusion provides weights for training on the imagenet. For LOL datasets, we still need to train ourselves, right?

Fayeben commented 1 year ago

For LOL dataset, we do not need to train the model. We still utilize the model from OpenAI / Guided_ Diffusion to restore the images.

Fayeben commented 1 year ago

Since there are no more questions, I'll mark it closed. Thanks!