Saving and Loading Model Weights PyTorch models store the learned parameters in an internal state dictionary, called ``state_dict``. These can be persisted via the ``torch.save`` method: model = models.vgg16(weights='IMAGENET1K_V1') torch.save(model.state_dict(), 'model_weights.pth') To load model weights, you need to create an instance of the same model first, and then load the parameters using ``load_state_dict()`` method.