'unexpected key "module.module.module.conv1.weight" in state_dict'

vision
8

It looks as if you’ve trained your model using nn.DataParallel and try to load it in a plain model.
Have a look at this thread for some solutions.

2 Likes