Struct OptimizerState 

pub struct OptimizerState {
    pub optimizer_type: String,
    pub version: String,
    pub param_groups: Vec<ParamGroupState>,
    pub state: HashMap<String, HashMap<String, Tensor>>,
    pub global_state: HashMap<String, f32>,
}

Optimizer state for serialization

Fields

optimizer_type: String

Optimizer type identifier

version: String

Version of the state format

param_groups: Vec<ParamGroupState>

Parameter group states

state: HashMap<String, HashMap<String, Tensor>>

Per-parameter optimizer state (keyed by parameter ID)

global_state: HashMap<String, f32>

Global optimizer state

Implementations

impl OptimizerState

pub fn new(optimizer_type: String) -> Self

Create a new empty optimizer state

pub fn validate(&self) -> Result<()>

Validate the state structure

pub fn total_param_count(&self) -> usize

Get the total number of parameters across all groups

pub fn is_compatible_with(&self, other: &OptimizerState) -> bool

Check if state is compatible with another state (same structure)

Trait Implementations

impl Clone for OptimizerState

fn clone(&self) -> OptimizerState

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizerState

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

Formats the value using the given formatter.