Unable to run demo.py(demo.py runtime error size mismatch for module.Prediction.weight)

[hic9507]

After learning, I try to test using demo.py using weights, but this error pops up and it does not run.
--character of train.py and demo.py are all unified to the same character, but still not executed.
please help
help me

Error
RuntimeError: Error(s) in loading state_dict for DataParallel:
size mismatch for module.Prediction.weight: copying a param with shape torch.Size([11520, 256]) from checkpoint, the shape in current model is torch.Size([11522, 256]).
size mismatch for module.Prediction.bias: copying a param with shape torch.Size([11520]) from checkpoint, the shape in current model is torch.Size([11522]).

[zaghdoud2019]

Hi,are you resolove the probleme please

[Benewend]

You need to open the demo file in the Paramus there is the character list juste make sure it is as yours