Using this code on my own dataset

[zhou-SHU164]

when i finished the training process, and came to the test, the error occured:

---

File "test.py", line 127, in
test()
File "test.py", line 41, in test
net.module.load_state_dict(torch.load(args.load_model)['net_state_dict'], strict = False) #got the net parameters of checkpoint
File "/home/**/anaconda3/envs/VRC/lib/python3.7/site-packages/torch/nn/modules/module.py", line 845, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Model:
size mismatch for decoder.encoder.sam_res1.conv1.weight: copying a param with shape torch.Size([64, 4, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 3, 1, 1]).
size mismatch for decoder.encoder.sam_res1.conv_res.weight: copying a param with shape torch.Size([64, 4, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 3, 1, 1]).

---

Maybe anyone could help me?