Struct CheckpointManager 

pub struct CheckpointManager<A: Float, D: Dimension> { /* private fields */ }

In-memory checkpoint manager

Implementations

impl<A: Float + ScalarOperand + Debug, D: Dimension + Send + Sync> CheckpointManager<A, D>

pub fn new() -> Self

Create a new checkpoint manager

pub fn with_max_checkpoints(_maxcheckpoints: usize) -> Self

Create a new checkpoint manager with maximum number of checkpoints

pub fn store_checkpoint( &mut self, key: String, checkpoint: OptimizerCheckpoint<A, D>, )

Store a checkpoint with a given key

pub fn get_checkpoint(&self, key: &str) -> Option<&OptimizerCheckpoint<A, D>>

Retrieve a checkpoint by key

pub fn remove_checkpoint( &mut self, key: &str, ) -> Option<OptimizerCheckpoint<A, D>>

Remove a checkpoint by key

pub fn list_checkpoints(&self) -> &[String]

List all checkpoint keys

pub fn clear(&mut self)

Clear all checkpoints

pub fn len(&self) -> usize

Get number of stored checkpoints

pub fn is_empty(&self) -> bool

Check if manager is empty

Trait Implementations

impl<A: Debug + Float, D: Debug + Dimension> Debug for CheckpointManager<A, D>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<A: Float + ScalarOperand + Debug, D: Dimension + Send + Sync> Default for CheckpointManager<A, D>

fn default() -> Self

Returns the “default value” for a type.