Skip to content

Engine / Checkpoint

Classes

Class Description
CheckpointConfigBase (abstract) Abstract base configuration for all checkpoint operations, holding the checkpoint format
CheckpointLoadConfig Full configuration for loading a checkpoint: path, format, and which state to restore
CheckpointLoadMetadataConfig Configuration for loading checkpoint metadata, controlling which config sections are loaded
CheckpointPathConfigBase (abstract) Abstract base configuration for checkpoint operations that require a filesystem path and optional timeout
CheckpointSaveConfig Full configuration for saving a checkpoint: path, format, weights, optimizer state, and file options
CheckpointSaveConfigBase (abstract) Abstract base configuration for saving checkpoints, with file-size and dtype options
CheckpointSaveMetadataConfig Configuration for saving checkpoint metadata (path and format) without weights or optimizer state
CheckpointStateConfigBase (abstract) Abstract base configuration for checkpoint operations that include model weights and/or optimizer state
CheckpointStateSaveConfigBase Configuration for saving model weights and/or optimizer state to a checkpoint