Checkpointing
Learn about running a classy pipeline for checkpointing and resuming training steps.
We'll cover the following
Saving checkpoints
To checkpoint the model to resume training later, we can use the save_checkpoint
method, which handles the state dictionaries for us and saves them to a file:
Get hands-on with 1400+ tech skills courses.