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
commented
4 years ago
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
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