Saving and loading a PyTorch model
One of the important aspects of working on neural network models is to save and load back a model after training. Think of a scenario where you have to make inferences from an already-trained model. You would load the trained model instead of training it again.
The following code can be found in the save_and_load_pytorch_model.ipynb file in the Chapter02 folder of this book’s GitHub repository at https://bit.ly/mcvp-2e.
Before going through the relevant commands to do that, taking the preceding example as our case, let’s understand what the important components that completely define a neural network are. We need the following:
- A unique name (key) for each tensor (parameter)
- The logic to connect every tensor in the network with one or the other
- The values (weight/bias values) of each tensor
While the first point is taken care of during the __init__ phase of a definition, the second...
