unexpected key "0.weight" in state_dict

last-one / Pytorch_Realtime_Multi-Person_Pose_Estimation

Pytorch version of Realtime Multi-Person Pose Estimation project

MIT License

216 stars 70 forks source link

unexpected key "0.weight" in state_dict #15

Closed CrossLee1 closed 6 years ago

CrossLee1 commented 6 years ago

I downloaded the pytorch model as stated in caffe2pytorch/README.md, but when I load the model in model.load_state_dict(state_dict), an error occurs: KeyError: 'unexpected key "0.weight" in 'state_dict'

Hope to get your response, thanks~

last-one commented 6 years ago

change 'name = k[7:]' to 'name = k'.