search

cszn / KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR
https://cszn.github.io/
MIT License
2.92k stars 627 forks source link

How can I train USRGan and other tiny models? #35

Open coreqode opened 4 years ago

coreqode commented 4 years ago

Hey, thanks for contributing. Your work is awesome. I tried to find the USRGan and USR-tiny models training support, but didn't get any of these? Is there any support for this currently?

cszn commented 4 years ago

If you get familiar with KAIR, you can modify it to train your own model.

coreqode commented 4 years ago

image I tried to train USRGan, using train_usrnet.json and main_train_msrresnet_gan.py/main_train_msrresnet_psnr.py. But it's showing, the above error. Look's like, the Discriminator net is not defined in json. What should I change?

FYJ10 commented 3 years ago

image I tried to train USRGan, using train_usrnet.json and main_train_msrresnet_gan.py/main_train_msrresnet_psnr.py. But it's showing, the above error. Look's like, the Discriminator net is not defined in json. What should I change?

I also want to run usrgan. Have you solved it?