TCMR missmatch??

[codeHorasan]

RuntimeError: Error(s) in loading state_dict for TCMR:
size mismatch for regressor.smpl.shapedirs: copying a param with shape torch.Size([6890, 3, 10]) from checkpoint, the shape in current model is torch.Size([6890, 3, 300]).

Anyone know about this?

[hongsukchoi]

I am not sure what your running (demo or test?), but just change the current model to [6890, 3, 10].

[codeHorasan]

I am running demo, python demo.py --vid_file demo.mp4 with CPU.

I am getting this error after loading pre-trained model from "tcmr_demo_model.pth.tar", at model.load_state_dict

ckpt = ckpt['gen_state_dict']
model.load_state_dict(ckpt, strict=False)

[hongsukchoi]

Version mismatch!
#16

You don't have to re-install or downgrade. Just change regressor.smpl.shapedirs to [6890,3,10] dimension

[codeHorasan]

Thank you for your help but I couldn't find on the code where I should change. It gives the error on TCMR, before even creating the SMPL model but in the error message, it mentions regressor.smpl.shapedirs. I tried to add "shapedirs = [6890,3,10]" to SMPL constructor but as I get the error before that line is compiled.

[hongsukchoi]

Then, the easiest way is to match the version. smplx==0.1.13

Or as I said, change the SMPL's shapedirs to 10 here

TCMR_RELEASE/lib/models/smpl.py

Line 61 in ade5ab5

class SMPL(_SMPL):

[ydl832]

Thank you for your help but I couldn't find on the code where I should change. It gives the error on TCMR, before even creating the SMPL model but in the error message, it mentions regressor.smpl.shapedirs. I tried to add "shapedirs = [6890,3,10]" to SMPL constructor but as I get the error before that line is compiled.

Hello, have you solved this problem? I have the same problem, do you know how to change the code? Thanks.