How many GPUs are needed for model training?

Xiangtaokong / MiOIR

Towards Effective Multiple-in-One Image Restoration: A Sequential and Prompt Learning Strategy

62 stars 2 forks source link

How many GPUs are needed for model training? #7

Open jiandandan001 opened 4 months ago

jiandandan001 commented 4 months ago

Thank you for sharing the code!

Which GPU was used for model training? How about the training time?

Is it possible to train this model using NVIDIA 3090 or 4090?

Thanks.

Xiangtaokong commented 4 months ago

Thanks for your attention. Different models require different GPUs. SwinIR needs 8 V100, others are usually need 2 V100. You can refer the .yml file in /options.