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.