Enum OptimizerKind 

pub enum OptimizerKind {
    Sgd {
        momentum: f32,
        nesterov: bool,
    },
    AdamW {
        beta1: f32,
        beta2: f32,
        eps: f32,
        weight_decay: f32,
    },
}

Variants

Sgd

Fields

momentum: f32

nesterov: bool

AdamW

Fields

beta1: f32

beta2: f32

eps: f32

weight_decay: f32

Trait Implementations

impl Clone for OptimizerKind

fn clone(&self) -> OptimizerKind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizerKind

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

Formats the value using the given formatter.

impl PartialEq for OptimizerKind

fn eq(&self, other: &OptimizerKind) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for OptimizerKind

impl StructuralPartialEq for OptimizerKind