ksuit.schedules.linear.decreasing

Classes

LinearDecreasingSchedule

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

Module Contents

class ksuit.schedules.linear.decreasing.LinearDecreasingSchedule(config)

Bases: ksuit.schedules.base.DecreasingProgressSchedule

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

Example

>>> schedule_config:
>>>   kind: ksuit.schedules.LinearDecreasingSchedule
>>>   max_value: ${model.optim.lr}
>>>   end_value: 0.0

Initialize the scheduler.

Parameters:

  • max_value – The initial and thus maximum value of the scheduler.

  • end_value – The final and thus minimum value of the scheduler.

  • config (ksuit.schemas.scheduler.DecreasingProgressScheduleConfig)