ksuit.schedules.polynomial.increasing¶

Classes¶

PolynomialIncreasingSchedule

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

Module Contents¶

class ksuit.schedules.polynomial.increasing.PolynomialIncreasingSchedule(config)¶

Bases: ksuit.schedules.base.IncreasingProgressSchedule

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

Initialize the scheduler.

Parameters:: config (ksuit.schemas.scheduler.PolynomialIncreasingScheduleConfig) – Configuration for the polynomial increasing schedule.

Example

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

power¶