Thanks for the quick answer.
The problem was that I wasn’t using torch.load… Since I wanted to save the state_dict along with some training/validation curves, I was saving a tuple (state_dict, results) using pickle…
I just tested and successfully saved a tuple like this with torch.save. I didn’t know this function was equivalent to pickle.dump ! Loading it back takes a split second.
Have a good day, sir!