Solved by Pranavan_Theivendira in post #2 
              
              
                Hi, 
The problem is that you are referring the same object. When you save it to the list instead of “snapshots.append(model.state_dict())”. Use the following command. 
snapshots.append(model.state_dict().copy())

Then only, you can save the values at that point. 
One small concern - when you ask qu…