class

ML::Optim::StepLR

Inherits ML::Optim::LRScheduler / Reference / Object

Step decay scheduler

Constructors

new(optimizer : Adam, step_size : Int32, gamma : Float32 = 0.1_f32)
Source

Instance methods

get_lr
Source
step
Source