Enum OptimizerType 

pub enum OptimizerType {
    SGD,
    SGDMomentum {
        momentum: u8,
    },
    Adam,
    AdamW {
        weight_decay: u8,
    },
}

Optimizer type for training

Variants

SGD

Stochastic Gradient Descent

SGDMomentum

SGD with momentum

Fields

momentum: u8

Momentum coefficient (typically 0.9)

Adam

Adam optimizer

AdamW

AdamW with weight decay

Fields

weight_decay: u8

Weight decay coefficient (as percentage, e.g., 1 = 0.01)

Trait Implementations

impl Clone for OptimizerType

fn clone(&self) -> OptimizerType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizerType

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

Formats the value using the given formatter.

impl Default for OptimizerType

fn default() -> OptimizerType

Returns the “default value” for a type.

impl PartialEq for OptimizerType

fn eq(&self, other: &OptimizerType) -> 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 OptimizerType

impl Eq for OptimizerType

impl StructuralPartialEq for OptimizerType