Solved by ptrblck in post #3 
              
              
                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.