You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Using GPU: 0
Load checkpoint from ./snapshot_7.pth.tar
Traceback (most recent call last):
File "demo/demo.py", line 75, in
model.load_state_dict(ckpt['network'], strict=False)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 845, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DataParallel:
size mismatch for module.human_model_layer.th_betas: copying a param with shape torch.Size([1, 10]) from checkpoint, the shape in current model is torch.Size([1, 300]).
size mismatch for module.human_model_layer.th_shapedirs: copying a param with shape torch.Size([6890, 3, 10]) from checkpoint, the shape in current model is torch.Size([6890, 3, 300]).
The text was updated successfully, but these errors were encountered:
vfr-xcugas
changed the title
help for sizemismatch problem
help for size mismatch problem
May 11, 2022
You are useing the wrong checkpoint file for example you are using basicModel_f_lbs_10_207_0_v1.1.0.pkl instead please use basicModel_f_lbs_10_207_0_v1.0.0.pkl, do same for male and Neutral
The text was updated successfully, but these errors were encountered: