search

naver / multi-hmr

Pytorch demo code and models for Multi-HMR
Other
158 stars 12 forks source link

Error with the new checkpoints #25

Closed sebastianopazo1 closed 2 weeks ago

sebastianopazo1 commented 2 weeks ago

Hi there! I was attempting to use the checkpoints to infer on some images, but I get the following error.

SMPLX found SMPL mean params is already here Loading model Traceback (most recent call last): File "/home/seba/Documents/multi-hmr/demo.py", line 283, in model = load_model(args.model_name) File "/home/seba/Documents/multi-hmr/demo.py", line 92, in load_model for k,v in vars(ckpt['args']).items(): KeyError: 'args' I'm using the same command, but with different checkpoint name Thank for your help!

syleemrl commented 2 weeks ago

I have the same problem. Key of new checkpoints is dict_keys(["state_dict_model") only, but those of older one are dict_keys(['epoch', 'iter', 'model_state_dict', 'args', 'scaler', 'best_val']).

fabienbaradel commented 2 weeks ago

Hi @syleemrl @sebastianopazo1 , thanks for checking the new checkpoints. Indeed the 'args' key with associated values was missing, sorry about that. I have just updated the new 672 checkpoints and it should work now. Please delete the ckpt and download them again from the website. It should solve the issue

syleemrl commented 2 weeks ago

@fabienbaradel It works! Thank you

sebastianopazo1 commented 2 weeks ago

Thanks @fabienbaradel !