Saving and Loading Models in the PyTorch tutorials

Sections What is a state_dict? and Saving & Loading a General Checkpoint for Inference and/or Resuming Training: separate model and optimizer dictionaries, stored training position, restoration, and training/evaluation mode.