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 + Send + Sync> OptimizerConfig<A>

pub fn new(lr: A) -> Self

Create a new optimizer configuration with the given learning rate

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

Set weight decay

pub fn grad_clip(self, gradclip: 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.

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 + Send + Sync> Default for OptimizerConfig<A>

fn default() -> Self

Returns the “default value” for a type.

impl<'de, A> Deserialize<'de> for OptimizerConfig<A>

where A: Deserialize<'de> + Float,

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<A> Serialize for OptimizerConfig<A>

where A: Serialize + Float,

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.