I am trying to load a pre-trained modal but for some reason, the dimensions are not setting current.

size mismatch for module.Prediction.weight: copying a param with shape torch.Size([50, 256]) from checkpoint, the shape in current model is torch.Size([37, 256]).
size mismatch for module.Prediction.bias: copying a param with shape torch.Size([50]) from checkpoint, the shape in current model is torch.Size([37]).

Can someone tell me what’s happening?