Solved by John_Smith in post #2 
              
              
                This issue has nothing to do with Load_state_dict(). The problem is when you created your model using MINIModel(a), you’ve set the reference of a as the data for the h variable via the following lines: 
    if not h0 is None:
        self.h.weight.data = h0

Basically, any change to self.h.weight.da…