noether.core.schedules.linear

Classes

LinearDecreasingScheduleConfig

LinearDecreasingSchedule

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

LinearIncreasingScheduleConfig

LinearIncreasingSchedule

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

Module Contents

class noether.core.schedules.linear.LinearDecreasingScheduleConfig(/, **data)

Bases: noether.core.schedules.base.DecreasingProgressScheduleConfig

Parameters:

data (Any)

kind: Literal['noether.core.schedules.LinearDecreasingSchedule'] = 'noether.core.schedules.LinearDecreasingSchedule'

The fully qualified class name of the scheduler.

class noether.core.schedules.linear.LinearDecreasingSchedule(config)

Bases: noether.core.schedules.base.DecreasingProgressSchedule

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

Example

schedule_config:
    kind: noether.core.schedules.LinearDecreasingSchedule
    max_value: ${model.optim.lr}
    end_value: 0.0

Initialize the scheduler.

Parameters:

config (noether.core.schedules.schemas.DecreasingProgressScheduleConfig)

class noether.core.schedules.linear.LinearIncreasingScheduleConfig(/, **data)

Bases: noether.core.schedules.base.IncreasingProgressScheduleConfig

Parameters:

data (Any)

kind: Literal['noether.core.schedules.LinearIncreasingSchedule'] = 'noether.core.schedules.LinearIncreasingSchedule'

The fully qualified class name of the scheduler.

class noether.core.schedules.linear.LinearIncreasingSchedule(config)

Bases: noether.core.schedules.base.IncreasingProgressSchedule

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

Example

schedule_config:
    kind: noether.core.schedules.LinearIncreasingSchedule
    max_value: ${model.optim.lr}
    start_value: 0.0

Initialize the scheduler.

Parameters:

config (noether.core.schedules.schemas.IncreasingProgressScheduleConfig)