When I use the pytorch model, I meet an error.

[Drsea123]

please! When I use the pytorch model, I meet an error.
I want to denoise an image using a one of the pretrained models

python test_ffdnet_ipol.py
--input input.png
--noise_sigma 25
--add_noise True

I have:

Testing FFDNet model

Parameters:
add_noise: True
input: 111.png
suffix:
noise_sigma: 0.09803921568627451
dont_save_results: False
no_gpu: False
cuda: True

rgb: False
im shape: (518, 774)
Loading model ...

Process finished with exit code 0
I have no use to get anything, but I have configured the libraries that need to be configured.
Then I debugged it and when I debugged this statement, the project exited.
if args['cuda']:
state_dict = torch.load(model_fn)
device_ids = [0]
model = nn.DataParallel(net, device_ids=device_ids).cuda()

I want to know why I can't use the training model?@cszn

[cszn]

https://github.com/cszn/KAIR