I have trouble when I test the pretrained model.

Question

I have trouble when I test the pretrained model.

Closed this issue 3 years ago · 2 comments

I follow the ReadMe to download the GOPRO_L1 pretrained model and unzip it to /experiment folder. However, when I test the pretrained model with a GOPRO blurry image, the output image is like:

How to deal with it? Many thanks.

Answer 1 · 2021-05-27T13:38:43.000Z

I have fixed this problem. My GPU is RTX3090 which does not match Pytorch v1.6 with cudatoolkit 10.2. So I use Pytorch v1.8 and get the right output image.

Answer 2 · 2021-05-28T03:08:41.000Z

Hi @daoyuli
Glad you solved the issue.
As far as I know, Ampere GPUs will perform the best with cuda >= 11.0