Struct CheckpointManager 

pub struct CheckpointManager {
    pub checkpoint_dir: PathBuf,
    pub format: CheckpointFormat,
    pub max_checkpoints: Option<usize>,
    pub naming_pattern: String,
}

Checkpoint manager for saving and loading model states

Fields

checkpoint_dir: PathBuf

Base directory for checkpoints

format: CheckpointFormat

Default checkpoint format

max_checkpoints: Option<usize>

Maximum number of checkpoints to keep

naming_pattern: String

Checkpoint naming pattern

Implementations

impl CheckpointManager

pub fn new<P: AsRef<Path>>(checkpoint_dir: P) -> NeuralResult<Self>

Create new checkpoint manager

pub fn with_format(self, format: CheckpointFormat) -> Self

Set checkpoint format

pub fn with_max_checkpoints(self, max_checkpoints: Option<usize>) -> Self

Set maximum number of checkpoints to keep

pub fn with_naming_pattern(self, pattern: String) -> Self

Set naming pattern

pub fn list_checkpoints(&self) -> NeuralResult<Vec<PathBuf>>

List available checkpoints

pub fn get_latest_checkpoint(&self) -> NeuralResult<Option<PathBuf>>

Get latest checkpoint