search

HRNet / Lite-HRNet

This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network.
Apache License 2.0
819 stars 126 forks source link

Why the shape between pretrianed weights and config is different? #60

Open xiguadong opened 3 years ago

xiguadong commented 3 years ago

the output info as belows: Use load_from_local loader The model and loaded state dict do not match exactly

size mismatch for keypoint_head.final_layer.weight: copying a param with shape torch.Size([16, 30, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 40, 1, 1]).

wangjing60755 commented 2 years ago

I meet the same error, how you solve this ?

wangjing60755 commented 2 years ago

I meet the same error, how you solve this ?

s94210252000 commented 2 years ago

Does anayone solve it?