Size mismatch error with pre-trained model

Question

Size mismatch error with pre-trained model

Closed this issue a year ago · 1 comments

Tonyzhang2000 commented a year ago

Hi, I got error message: "size mismatch for extension_layer.hand_contact_state_layer.3.weight: copying a param with shape torch.Size([5, 32]) from checkpoint, the shape in current model is torch.Size([4, 32])." when running your code with provided pre-trained model. A temporary work around would be checkout to previous version.

Answer 1 · 2023-03-08T22:20:45.000Z

Hi, thanks for pointing it out. I accidentally pushed a commit containing some of my trials of another experiment. I have reverted it back and everything should work now.