Missing keys & unexpected keys in state_dict when loading self trained model

[ptrblck]

Could you try to load the state_dict using model.load_state_dict(torch.load(PATH), strict=False)?