I have the following error:
RuntimeError: While copying the parameter named loss.weight, whose dimensions in the model are torch.Size([20]) and whose dimensions in the checkpoint are torch.Size([20]).

If the sizes agree, then what can be the issue ?