Saving and Loading Models in the PyTorch tutorials
= Saving and Loading Models in the PyTorch tutorials
{c}
https://docs.pytorch.org/tutorials/beginner/saving_loading_models.html
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.