In this post , we will be seeing about how to save and load models in pytorch

Why to Save Model ?

Despite the framework that is being used, saving the model is a very important thing so as to use it again readily instead of training the neural network again from the start.

Sometimes, training might even take weeks to be completed. So, each and everytime we need to use our network, we can’t afford to train them right from the beginning.

another advantage of saving the models then and there is that we can compare 2 different models to choose which in is much more effective and efficient for the task given.

having said that, let’s jump right into the code

#pytorch #pytorch-tutorial #saving-the-model #torch-save #deep-learning