Enum LearningRate 

pub enum LearningRate {
    Constant(f32),
    Schedule(Arc<dyn Fn(&Array) -> Array + Send + Sync>),
}

A learning rate: a constant or a function of the step (a 0-D array, from 0) returning a 0-D array, so a compiled step sees every new rate.

Variants

Constant(f32)

The same rate at every step (it initialises the state’s rate array).

Schedule(Arc<dyn Fn(&Array) -> Array + Send + Sync>)

The rate at each step, written into the state before each update.

Implementations

impl LearningRate

pub fn at(&self, step: &Array) -> Array

The rate at step (a 0-D array), in float32.

Trait Implementations

impl Clone for LearningRate

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LearningRate

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

Formats the value using the given formatter.

impl From<f32> for LearningRate

fn from(lr: f32) -> Self

Converts to this type from the input type.

impl From<f64> for LearningRate

fn from(lr: f64) -> Self

Converts to this type from the input type.