key errors: unexpected keys in state_dict
longzee opened this issue · 1 comments
longzee commented
Hi,zihua:
after training , I run predict.py
, then I get an error shows that keys are mismatch. Could you help me solve this bug? Thanks ~
The errror information:
RuntimeError: Error(s) in loading state_dict for VGG:
Missing key(s) in state_dict: "features.0.weight", "features.0.bias"...
Unexpected key(s) in state_dict: "module.features.0.weight", "module.features.0.bias"...