Solved by AlphaBetaGamma96 in post #11 
              
              
                Ah my apologises, I should’ve phrased the last statement more clearly. I meant to try the for key, value in state_dict expression for your original torch.save object. It makes sense it requires model_state_dict as that’s the key we use to save the model’s state_dict! 
When loading the model you’re u…