Solved by smth in post #4 
              
              
                if you saved the state dict, you have to load the model via load_state_dict. Either read the documentation or look at our examples.