JingyunLiang / SwinIR

SwinIR: Image Restoration Using Swin Transformer (official repository)

Home Page:https://arxiv.org/abs/2108.10257

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About training code

Freeman-Cai opened this issue · comments

I have download the training code from https://github.com/cszn/KAIR but it generates the G and E models when i try to do a SR task. Do I download the wrong code?

No. It's correct. G and E means the commonly used Generator and the Generator_EMA (the interpolation between models of the last iteration and the current iteration. It leads to more stable model training.)

Thanks for your reply! I will go on to understand the code.

Feel free to open it if you have more questions.