Missing keys in load_state_dict

Hi all,

I finished training the model and I’m trying to load it for evaluation at the moment. However, load_state_dict raise error:

Missing key(s) in state_dict: "iter", "losses.class_map.cls_linear.weight", ...

But when I tried to print the state_dict.keys(), the keys are the same as the missing keys:

odict_keys(['iter', 'losses.class_map.cls_linear.weight', ...

The code for the model is quite big so I can’t paste it here.

Did anyone encounter this bug? Any idea of what might be causing this?

Any help is appreciated. Thanks.