noether.core.schedules.polynomial¶

Classes¶

`PolynomialDecreasingScheduleConfig`
`PolynomialDecreasingSchedule`	A scheduler that decreases polynomially from the maximum to minimum value over the total number of steps.
`PolynomialIncreasingScheduleConfig`
`PolynomialIncreasingSchedule`	A scheduler that increases polynomially from the minimum to maximum value over the total number of steps.

Module Contents¶

class noether.core.schedules.polynomial.PolynomialDecreasingScheduleConfig(/, **data)¶

Bases: noether.core.schedules.schemas.DecreasingProgressScheduleConfig

Parameters:: data (Any)

kind: Literal['noether.core.schedules.PolynomialDecreasingSchedule'] = 'noether.core.schedules.PolynomialDecreasingSchedule'¶: The fully qualified class name of the scheduler.

power: float = None¶: The power of the polynomial function.

class noether.core.schedules.polynomial.PolynomialDecreasingSchedule(config)¶

Bases: noether.core.schedules.base.DecreasingProgressSchedule

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

Parameters:: config (PolynomialDecreasingScheduleConfig) – Configuration for the polynomial decreasing schedule. See PolynomialDecreasingScheduleConfig for details.

Example

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

power¶

class noether.core.schedules.polynomial.PolynomialIncreasingScheduleConfig(/, **data)¶

Bases: noether.core.schedules.schemas.IncreasingProgressScheduleConfig

Parameters:: data (Any)

kind: Literal['noether.core.schedules.PolynomialIncreasingSchedule'] = 'noether.core.schedules.PolynomialIncreasingSchedule'¶: The fully qualified class name of the scheduler.

power: float = None¶: The power of the polynomial function.

class noether.core.schedules.polynomial.PolynomialIncreasingSchedule(config)¶

Bases: noether.core.schedules.base.IncreasingProgressSchedule

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

Parameters:: config (PolynomialIncreasingScheduleConfig) – Configuration for the polynomial increasing schedule. See PolynomialIncreasingScheduleConfig for details.

Example

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

power¶