I was reading Save and Load model 45 but it wasn’t clear why I’d use torch.save over pickle.dump.
What worries me is that my Neural Net modules/object have many more things inside of them besides only parameters. Since pytorch saves things using the state_dict that worried me that something might be missing (https://pytorch.org/tutorials/beginner/saving_loading_models.html 24).
Isn’t for my use case easier to simply use the pickle.dump function? Why would I use torch.save in general?
Note I am using gpu’s. Not sure if that matters.
The easiest thing for me is if I could just save the whole python program state (similar to how matlab used to allow you to do it https://www.mathworks.com/help/matlab/ref/save.html 7).