ROMP icon indicating copy to clipboard operation
ROMP copied to clipboard

Dimension mismatch for trace_head.pkl

Open GuanlinWU123 opened this issue 1 year ago • 2 comments

Hi! When calling the temporal_head_model = load_model(self.trace_head_model_path, temporal_head_model, prefix='', drop_prefix='', fix_loaded=False), I encounter a problem that "copy param _result_parser.params_map_parser.smpl_model.vertex_joint_selector.extra_joints_idxs failed, mismatched Error message: The size of tensor a (11) must match the size of tensor b (21) at non-singleton dimension 0 Dimension mismatch for '_result_parser.params_map_parser.smpl_model.vertex_joint_selector.extra_joints_idxs': current model - torch.Size([11]), loaded state dict - torch.Size([21])" I think it should be the problem of the dimension in trained model trace_head.pkl. Is there any solution to solve this problem?

GuanlinWU123 avatar Mar 14 '24 05:03 GuanlinWU123

Hi! I also encountered the same problem. How do you solve this?

IridescentJiang avatar Feb 11 '25 11:02 IridescentJiang

The same problem.

Dan1900 avatar Sep 15 '25 06:09 Dan1900