search

hassony2 / obman_train

[cvpr19] Demo, training and evaluation code for generating dense hand+object reconstructions from single rgb images
https://hassony2.github.io/obman.html
GNU General Public License v3.0
188 stars 27 forks source link

Missing states when loading model for Image Demo #34

Open SupreethN opened 2 years ago

SupreethN commented 2 years ago

Thanks for open sourcing this work.

I am trying to run a demo on a single image using the command: python image_demo.py --resume release_models/obman/checkpoint.pth.tar

However, I get the following error:

RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.mano_branch.mano_layer_right.th_comps", "module.mano_branch.mano_layer_left.th_comps".

I see that the model has 2 additional layers that are not present in the checkpoint state dictionary.

Can you please help me with this? What am I missing here? The MANO models I am using are mano_v1_2.