Solved by user_123454321 in post #4 
              
              
                Ah…so the model is not saved as dictionary, but the object itself is saved ? Try this… 
model = torch.load(model_path, map_location='cpu')
model = model.module