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.95k stars 630 forks source link

dear writer, how to use gpu to run imdn? #19

Closed I-believe-sunshine closed 4 years ago

cszn commented 4 years ago

https://github.com/cszn/KAIR/blob/9e65edabefaf0331c1cc526d533d779e47824b00/main_test_imdn.py#L102