It seems you’ve used nn.DataParallel
to save the model.
You could wrap your current model again in nn.DataParallel
or just remove the .module
keys.
Here is a similar thread with some suggestions.
33 Likes