Solved by justusschock in post #2 
              
              
                You have to create a model instance and then load the saved weights as statdict: 
model = MyModel() 
model.load_state_dict(torch.load('model_best.pth.tar')['state_dict'])

The statedict itself is only a dict containing the tensor names and the corresponding weights. It has no information of the mode…