I have a PyTorch model that has test accuracy of about 97%. I save it using torch.save(my_model.state_dict(), PATH) , but whenever I try to reload it using my_model.load_state_dict(torch.load(PATH)) and test it on the same data using test_fn(my_model) my test accuracy goes down to about 0.06%. The same thing happens if I use test_fn(my_model.eval()) . Is there an extra step I need to take?