Enum LrSchedule 

pub enum LrSchedule {
    Constant,
    Step {
        step_size: usize,
        gamma: f32,
    },
    CosineAnnealing {
        min_lr: f32,
    },
}

Learning rate schedule (applied per epoch).

Variants

Constant

Keep lr constant.

Step

Step decay: lr = lr0 * gamma^(epoch / step_size).

Fields

step_size: usize

gamma: f32

CosineAnnealing

Cosine annealing from lr0 down to min_lr.

Fields

min_lr: f32

Implementations

impl LrSchedule

pub fn validate(self) -> Result<()>

Trait Implementations

impl Clone for LrSchedule

fn clone(&self) -> LrSchedule

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LrSchedule

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

Formats the value using the given formatter.

impl Default for LrSchedule

fn default() -> LrSchedule

Returns the “default value” for a type.

impl PartialEq for LrSchedule

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

impl StructuralPartialEq for LrSchedule