Struct ParamGroupState 

pub struct ParamGroupState {
    pub lr: f32,
    pub options: HashMap<String, f32>,
    pub param_count: usize,
}

Parameter group state

Fields

lr: f32

Learning rate for this group

options: HashMap<String, f32>

Additional options for this group

param_count: usize

Number of parameters in this group (for validation)

Implementations

impl ParamGroupState

pub fn new(lr: f32, param_count: usize) -> Self

Create a new parameter group state

pub fn from_param_group(group: &ParamGroup) -> Self

Create from a ParamGroup

pub fn get_option(&self, key: &str, default: f32) -> f32

Get an option value with a default

pub fn set_option(&mut self, key: String, value: f32)

Set an option value

Trait Implementations

impl Clone for ParamGroupState

fn clone(&self) -> ParamGroupState

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ParamGroupState

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

Formats the value using the given formatter.