Struct OptimizerCheckpoint 

pub struct OptimizerCheckpoint {
    pub step: u64,
    pub epoch: u32,
    pub optimizer_name: String,
    pub param_states: HashMap<String, ParamState>,
    pub hyperparams: HashMap<String, f64>,
    pub metadata: CheckpointMetadata,
}

Serialisable snapshot of an optimizer’s full training state.

Fields

step: u64

epoch: u32

optimizer_name: String

param_states: HashMap<String, ParamState>

Parameter name → per-parameter state.

hyperparams: HashMap<String, f64>

metadata: CheckpointMetadata

Implementations

impl OptimizerCheckpoint

pub fn new(optimizer_name: impl Into<String>, step: u64, epoch: u32) -> Self

Create an empty checkpoint for the given optimizer at step/epoch.

pub fn add_param_state(&mut self, name: impl Into<String>, state: ParamState)

Insert or replace the state for one parameter.

pub fn set_hyperparam(&mut self, key: impl Into<String>, value: f64)

Record a scalar hyper-parameter (learning rate, beta1, etc.).

pub fn get_hyperparam(&self, key: &str) -> Option<f64>

Retrieve a previously recorded hyper-parameter value.

pub fn num_params(&self) -> usize

Number of parameters stored in this checkpoint.

pub fn total_elements(&self) -> usize

Total number of scalar elements across all first-moment vectors.

Trait Implementations

impl Clone for OptimizerCheckpoint

fn clone(&self) -> OptimizerCheckpoint

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizerCheckpoint

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for OptimizerCheckpoint

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for OptimizerCheckpoint

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.