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 1200+ tech skills courses.