size mismatch error

[afsharshamsi]

Dear VITpose authors, I decided to run your code on mpii dataset using ViTPose-B. after downloading the checkpoint and config for that, I got the size mismatch error as "size mismatch for keypoint_head.final_layer.weight: copying a param with shape torch.Size([17, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 256, 1, 1]).
size mismatch for keypoint_head.final_layer.bias: copying a param with shape torch.Size([17]) from checkpoint, the shape in current model is torch.Size([16]).". It seems the provided checkpoint does not match leading to the following results "Ankle: 0.6140770714586143
Elbow: 2.4202130750155133
Head: 4.058663028649386
Hip: 0.32892339747001
Knee: 89.34101785856382
PCKh: 13.827738745771532
PCKh@0.1: 3.851157936458084
Shoulder: 0.18682065217391303
Wrist: 9.593513488382577". This results are far from what you reported in the paper. Can you please provide me with the correct checkpoint as well as config please?