Copying a param with shape torch. Size ([262, 2048]), parameter size does not match

A parameter with shape torch. Size ([262]) is copied from the checkpoint, and the shape in the current model is torch. Size ([290]).

The parameter size of fc.weight does not match, just modify the parameter.

VIM open the corresponding file and modify the parameters

Solve the problem successfully

Reference link

 

Read More: