Struct ModelCheckpoint 

pub struct ModelCheckpoint {
    pub parameters: HashMap<String, Tensor>,
    pub metadata: ModelMetadata,
    pub optimizer_state: Option<HashMap<String, Vec<f32>>>,
}

Model checkpoint containing parameters and metadata

Fields

parameters: HashMap<String, Tensor>

Model parameters (weights and biases)

metadata: ModelMetadata

Model metadata

optimizer_state: Option<HashMap<String, Vec<f32>>>

Optimizer state (optional)

Implementations

impl ModelCheckpoint

pub fn new(parameters: HashMap<String, Tensor>) -> Self

Create a new checkpoint

pub fn with_metadata(self, metadata: ModelMetadata) -> Self

Set metadata

pub fn with_optimizer_state(self, state: HashMap<String, Vec<f32>>) -> Self

Set optimizer state

pub fn save<P: AsRef<Path>>(&self, path: P) -> Result<()>

Save checkpoint to file

pub fn load<P: AsRef<Path>>(path: P) -> Result<Self>

Load checkpoint from file

Trait Implementations

impl Clone for ModelCheckpoint

fn clone(&self) -> ModelCheckpoint

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelCheckpoint

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

Formats the value using the given formatter.