cszn
When I use the pytorch model, I meet an error.
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
