neuralqx.optimizer.schedule module¶

build_lr(spec)¶

Constant learning rate.

build_lr(spec: neuralqx.optimizer.types.ExponentialDecay) → neuralqx.optimizer.types.OptaxSchedule

build_lr(spec: neuralqx.optimizer.types.CosineDecay) → neuralqx.optimizer.types.OptaxSchedule

build_lr(spec: neuralqx.optimizer.types.LinearDecay) → neuralqx.optimizer.types.OptaxSchedule

Return type:: float

make_learning_rate(spec)¶

Return type:: Union[float, OptaxSchedule]