Learn how to use PyTorch packages to prepare and load common datasets for your model.
Learn how to use PyTorch's torch.nn package to create and define a neural network the MNIST dataset.
Learn how state_dict objects, Python dictionaries, are used in saving or loading models from PyTorch.
Learn about the two approaches for saving and loading models for inference in PyTorch - via the state_dict and via the entire model.
Saving and loading a general checkpoint model for inference or resuming training can be helpful for picking up where you last left off. In this recipe, explore how to save and load multiple checkpoints.
In this recipe, learn how saving and loading multiple models can be helpful for reusing models that you have previously trained.