Deploying / Making Predictions

Using our model for making predictions

Again, if we are starting fresh as if we had just turned on the computer and started Jupyter, we have to set up the stage before actually loading the model. But, this time, we only need to configure the model:

Once again, we have an untrained model at this point. The loading procedure is simpler though:

• Load the dictionary back by using torch.load().

• Load model state dictionary back by using its method load_state_dict().

Since the model is fully trained, we do not need to load the optimizer or ...