ml.lr_schedulers.constant

Defines a constant learning rate scheduler.

class ml.lr_schedulers.constant.ConstantLRSchedulerConfig(name: str = '???', factor: float = 1.0)[source]

class ml.lr_schedulers.constant.ConstantLRScheduler(config: BaseConfigT)[source]

get_lr_scale(state: State) → float[source]

Given a state, returns the current learning rate.