how to save a neural network pytorch

Saving:
	torch.save(model, PATH)


Loading: 
	model = torch.load(PATH)
	model.eval()

Here is what the above code is Doing:
1. Loads the model from the path specified.
2. Sets the model to evaluation mode.
3. Loads the image from the path specified.
4. Resizes the image to the size specified.
5. Converts the image to a PyTorch tensor.
6. Passes the image through the model to get a prediction.
7. Takes the highest predicted class index.
8. Looks up the class name in the class index.
9. Prints the class name and the probability of the prediction.