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

[ptrblck]

Could you post the model definition, please?

@lelegu

If I don’t use nn.DataParallel to save model, I also don’t need it when I load it, right?

Yes, that is correct.