Dimension mismatch for trace_head.pkl

[GuanlinWU123]

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?