Solved by ptrblck in post #2 
              
              
                It seems that model_to_save['model_state'] contains a state_dict, which is the recommended way to serialize the models. 
To restore the model, create a model instance first and load the corresponding state_dict afterwards. 
I would not recommend to save the model directly via torch.save(model) but t…