Skip to main content

Module checkpoint_v2

Module checkpoint_v2 

Source
Expand description

Versioned tensor checkpoint manager with delta compression.

TensorCheckpointManagerV2 manages a history of named tensor checkpoints, automatically selecting between full snapshots and delta-compressed entries based on a configurable CheckpointPolicy. Delta entries record only the tensors that changed relative to the previous checkpoint, enabling efficient storage of model evolution in federated learning.

Structsยง

CheckpointDelta
Describes the incremental change between two checkpoint versions.
CheckpointEntry
A single checkpoint record stored by the manager.
CheckpointPolicy
Controls eviction and storage-format decisions made by the manager.
CheckpointStats
Aggregate statistics maintained by the manager.
TensorCheckpointManagerV2
Versioned tensor checkpoint manager with delta compression.