Checkpointing

Learn about running a classy pipeline for checkpointing and resuming training steps.

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.