ksuit.schedules.polynomial.decreasing¶

Classes¶

PolynomialDecreasingSchedule

A scheduler that decreases polynomially from the maximum to minimum value over the total number of steps.

Module Contents¶

class ksuit.schedules.polynomial.decreasing.PolynomialDecreasingSchedule(config)¶

Bases: ksuit.schedules.base.DecreasingProgressSchedule

A scheduler that decreases polynomially from the maximum to minimum value over the total number of steps.

Initialize the scheduler.

Parameters:: config (ksuit.schemas.scheduler.PolynomialDecreasingScheduleConfig) – Configuration for the polynomial decreasing schedule.

Example

>>> schedule_config:
>>>     kind: ksuit.schedules.PolynomialDecreasingSchedule
>>>     power: 2.0
>>>     start_value: ${model.optim.lr} # reference to the lr defined above
>>>     end_value: 1e-6

power¶