KAIR icon indicating copy to clipboard operation
KAIR copied to clipboard

How can I train USRGan and other tiny models?

Open coreqode opened this issue 4 years ago • 3 comments

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?

coreqode avatar Sep 19 '20 11:09 coreqode

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

cszn avatar Sep 19 '20 13:09 cszn

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?

coreqode avatar Sep 22 '20 18:09 coreqode

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?

FYJ10 avatar Sep 09 '21 01:09 FYJ10