search

assafshocher / ZSSR

"Zero-Shot" Super-Resolution using Deep Internal Learning
Other
401 stars 102 forks source link

KernelGAN + ZSSR #22

Open jiandandan001 opened 4 years ago

jiandandan001 commented 4 years ago

Hi, Thank you for sharing the code.

I am running the KernelGAN + ZSSR. It seems that the ZSSR part runs on the CPU, so it is very time-consuming. How can I run it on the GPU?

The enviroment is as follows: windows 10, pytorch 1.0, python 3.6. Thanks.

assafshocher commented 4 years ago

ZSSR runs on GPU if defined to do so. See the readme regarding specification of the GPU num.

jiandandan001 commented 4 years ago

ZSSR runs on GPU if defined to do so. See the readme regarding specification of the GPU num.

Thank you for your reply.

For the ZSSR, I tried the command 'python run_ZSSR.py ' . However, I found it still runs on CPU. Do you have any suggestions, thanks.