Error in loading state_dict for PoseResNet

xingyizhou / pytorch-pose-hg-3d

PyTorch implementation for 3D human pose estimation

GNU General Public License v3.0

615 stars 141 forks source link

Error in loading state_dict for PoseResNet #55

Closed Peter654q closed 5 years ago

Peter654q commented 5 years ago

Hi, I want to use the weight I trained in stage 1 and continue training in stage 2, but it has runtime error. RuntimeError: Error(s) in loading state_dict for PoseResNet: Missing key(s) in state_dict: "depth.weight", "depth.bias". Do you have any idea can solve this problem?

Thanks!

xingyizhou commented 5 years ago

Hi, Thanks for the report! I have fixed the problem https://github.com/xingyizhou/pytorch-pose-hg-3d/commit/b8afd1027f5e846e5923ac8deefafc76869ed788#diff-30a30f8b61024bfe24b090332ccda819R26 .

Peter654q commented 5 years ago

Thanks for your reply!