Struct ParamStateSnapshot 

pub struct ParamStateSnapshot {
    pub param_name: String,
    pub step: u64,
    pub first_moment: Option<Vec<f64>>,
    pub second_moment: Option<Vec<f64>>,
    pub momentum: Option<Vec<f64>>,
    pub ema: Option<Vec<f64>>,
    pub grad_norm: Option<f64>,
}

A snapshot of optimizer state for one parameter.

Fields

param_name: String

step: u64

first_moment: Option<Vec<f64>>

First moment (m) — Adam exponential moving average of gradient.

second_moment: Option<Vec<f64>>

Second moment (v) — Adam exponential moving average of squared gradient.

momentum: Option<Vec<f64>>

SGD momentum buffer.

ema: Option<Vec<f64>>

EMA weights for Lion optimizer.

grad_norm: Option<f64>

Last observed gradient norm.

Implementations

impl ParamStateSnapshot

pub fn new(param_name: impl Into<String>) -> Self

Create a new empty snapshot for the named parameter.

Trait Implementations

impl Clone for ParamStateSnapshot

fn clone(&self) -> ParamStateSnapshot

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ParamStateSnapshot

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

Formats the value using the given formatter.