HVision-NKU/SRFormer

Question about training machine.

mls1999725 opened this issue · 2 comments

Thanks for your excellent work! I would like to ask how many GPUs are used for training SRFormer and SwinIR? What type of GPUs are they? Looking forward to your reply.

Hi!
For the official setting, you may require two 3090 GPUs to train SRFormer or SwinIR. You can decrease the batch size, window size, patch size, and other hyper-parameters to train it on less GPU. I encourage you to try to configure these hyper-parameters!

We will close the issue since it has been inactive for a long time. Feel free to reopen it when you need.