Solved by ptrblck in post #2 
              
              
                If you’ve saved the state_dict you can load it with model.load_state_dict(torch.load(PATH_TO_DICT)). 
Have a look at the Serialization Tutorial for more information.