Struct OptimizerConfig

pub struct OptimizerConfig<A: Float> {
    pub lr: A,
    pub weight_decay: A,
    pub grad_clip: Option<A>,
    pub params: HashMap<String, A>,
}

Unified optimizer configuration

Fields

lr: A

Learning rate

weight_decay: A

Weight decay (L2 regularization)

grad_clip: Option<A>

Gradient clipping value (optional)

params: HashMap<String, A>

Additional optimizer-specific parameters

Implementations

impl<A: Float> OptimizerConfig<A>

pub fn new(lr: A) -> Self

Create a new optimizer configuration with the given learning rate

pub fn weight_decay(self, weight_decay: A) -> Self

Set weight decay

pub fn grad_clip(self, grad_clip: A) -> Self

Set gradient clipping

pub fn param<S: Into<String>>(self, key: S, value: A) -> Self

Add a custom parameter

pub fn params(self, params: HashMap<String, A>) -> Self

Set multiple parameters at once

Trait Implementations

impl<A: Clone + Float> Clone for OptimizerConfig<A>

fn clone(&self) -> OptimizerConfig<A>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<A: Debug + Float> Debug for OptimizerConfig<A>

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

Formats the value using the given formatter.

impl<A: Float> Default for OptimizerConfig<A>

fn default() -> Self

Returns the “default value” for a type.