Trainer.load_checkpoint¶

Trainer.load_checkpoint(checkpoint_manager, step=None, best=False)[source]¶

Loads a checkpoint from the checkpoint manager.

Parameters:

checkpoint_manager (CheckpointManager) – Checkpoint manager to use for loading the checkpoint.
step (int | None) – Step of the checkpoint to load. If None, load the latest checkpoint.
best (bool) – If true, load the best checkpoint.