noether.core.schedules.polynomial

Classes

PolynomialDecreasingSchedule

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

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.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 (noether.core.schemas.schedules.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.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 (noether.core.schemas.schedules.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