load state_dict

[amkmpro]

Thank you for your code.
I want to use MobileNet2. I trained it (with e.g 10 epochs) and I save state_dict, but when I use it, I have an error.
Error is:
RuntimeError: Error(s) in loading state_dict for MobileNetV2:
Missing key(s) in state_dict: "conv1.weight", ...
Unexpected key(s) in state_dict: "module.conv1.weight", ....
Please help me.
Thanks in advance.

[amkmpro]

I found a solution. I use
torch.save(net.module.state_dict(), PATH)
instead of
torch.save(net.state_dict(), PATH)