Missing states when loading model for Image Demo

[SupreethN]

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.