Struct OptimizerOptions 

pub struct OptimizerOptions {
    pub lr: f32,
    pub weight_decay: f32,
    pub eps: f32,
    pub maximize: bool,
}

Common optimizer options

Fields

lr: f32

weight_decay: f32

eps: f32

maximize: bool

Implementations

impl OptimizerOptions

pub fn new(lr: f32) -> Self

Create new optimizer options with specified learning rate

pub fn with_weight_decay(self, weight_decay: f32) -> Self

Set weight decay

pub fn with_eps(self, eps: f32) -> Self

Set epsilon value for numerical stability

pub fn with_maximize(self, maximize: bool) -> Self

Set maximize flag (for maximization problems)

pub fn to_hashmap(&self) -> HashMap<String, f32>

Convert to HashMap for compatibility with parameter groups

pub fn from_hashmap(map: &HashMap<String, f32>) -> Self

Create from HashMap

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

Validate the options are reasonable

pub fn create_standard_state_dict( optimizer_type: &str, version: Option<&str>, param_groups: &[ParamGroup], state: &HashMap<String, HashMap<String, Tensor>>, global_state: Option<HashMap<String, f32>>, ) -> OptimizerState

Create standardized state dict for any optimizer

pub fn validate_state_compatibility( current_groups: &[ParamGroup], state_groups: &[ParamGroupState], ) -> Result<()>

Validate state dict compatibility between optimizers

Trait Implementations

impl Clone for OptimizerOptions

fn clone(&self) -> OptimizerOptions

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizerOptions

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

Formats the value using the given formatter.

impl Default for OptimizerOptions

fn default() -> Self

Returns the “default value” for a type.