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

[Tavados]

In my case it would not throw any errors anymore, but it wouldn’t correctly load the state_dict either.
In the end I just had to do this to just remove the “.module” part.