Test model problem

[wenfan-wenfan]

Dear Mr. Author@cszn, the work of DPSR is of great significance and gives me great inspiration. when I fully followed the requirements of your training environment and used the "main_train_dpsr.py" provided by you to train the model, I put the generated model into the "main_test_dpsr.py", but reported the following Error:
RuntimeError: Error(s) in loading state_dict for SRResNet:
Missing key(s) in state_dict: "model.2. Weight ", "model.2. Bias ", "model.5. Weight ", "model.5.
Unexpected key(s) in state_dict: "model.3. Weight ", "model.3. Bias ", "model.6. Weight ", "model

When I use your trained "dpsr_x2.pth","dpsr_x3.pth" and "dpsr_x4.pth" to check, but there is no problem, I have tried many ways in the inquiry and failed to solve, Do you have any good suggestion or solutions to my problem

this is the error picture：phttps://imgchr.com/i/JjAZVK

[cszn]

You should also change

DPSR/models/network_srresnet.py

Line 13 in 2ec93c7

class SRResNet(nn.Module):

[wenfan-wenfan]

You should also change

DPSR/models/network_srresnet.py

Line 13 in 2ec93c7

class SRResNet(nn.Module):

I'm sorry to disturb you again. Today, I changed “class SRResNet (nn.Module): ” in network_srresnet.py to network_dpsr.py as you suggested, but I still reported an error. In addition, I still followed the practice of many people on the Internet to add
model = nn.DataParallel(model)
in front of
model.load_state_dict(torch.load(model_path), strict=False)

, but there is still no correct result, so could you please tell me a detailed solution to this problem?Once again, I would like to express my heartfelt thanks to you！