superbrucejia / nlnet-iqa Goto Github PK

Hi there, I am facing issue with real_testing.py throwing error, I have removed cuda() from Model as I don't have GPU, also I have added map_location=torch.device('cpu') to torch.load:

cmd used: python real_testing.py --model_file save_model/TID2013-32-4-1.pth --im_path 'test_images/cr7.jpg'

Models Used: Multiple TID, train*, LIVE

Error:
Traceback (most recent call last):
File "/home/AI/NLNet-IQA/real_testing.py", line 125, in
run(args)
File "/home/AI/NLNet-IQA/real_testing.py", line 35, in run
model.load_state_dict(torch.load(args.model_file,map_location=torch.device('cpu')))
File "/opt/intel/oneapi/intelpython/latest/envs/pytorch-gpu/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for Network:
size mismatch for dis_type_linear2.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([6, 512]).
size mismatch for dis_type_linear2.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([6]).