mlx_rs::optimizers

Type Alias AdafactorBuilderLr

Source
pub type AdafactorBuilderLr = Option<f32>;
Expand description

Option<Array>. Type alias for the learning rate used in Adafactor builder due to limitation in the generate_builder macro

Aliased Type§

enum AdafactorBuilderLr {
    None,
    Some(f32),
}

Variants§

§1.0.0

None

No value.

§1.0.0

Some(f32)

Some value of type T.