Struct dfdx::optim::RMSpropConfig

pub struct RMSpropConfig {
    pub lr: f32,
    pub alpha: f32,
    pub eps: f32,
    pub momentum: Option<f32>,
    pub centered: bool,
    pub weight_decay: Option<WeightDecay>,
}

Configuration of hyperparameters for RMSprop.

Fields

lr: f32

Learning rate. Defaults to 1e-2.

alpha: f32

Value for exponential moving average. Defaults to 0.9.

eps: f32

Epsilon for stability. Defaults to 1e-8.

momentum: Option<f32>

Optional momentum. Defaults to None.

centered: bool

Whether the avg should be centered by the grad’s avg value. Defaults to false.

weight_decay: Option<WeightDecay>

Optional weight decay. Defaults to None.

Trait Implementations

impl Clone for RMSpropConfig

fn clone(&self) -> RMSpropConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for RMSpropConfig

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

Formats the value using the given formatter.

impl Default for RMSpropConfig

fn default() -> Self

Returns the “default value” for a type.

impl Copy for RMSpropConfig