ksuit.schedules.base.sequential_step_schedule

Classes

SequentialStepScheduleConfig
SequentialStepSchedule	A scheduler that switches between multiple schedules based on the step number.

Module Contents

class ksuit.schedules.base.sequential_step_schedule.SequentialStepScheduleConfig

schedule: ksuit.schedules.base.schedule_base.ScheduleBase

start_step: int = None

end_step: int = None

class ksuit.schedules.base.sequential_step_schedule.SequentialStepSchedule(schedule_configs, **kwargs)

Bases: ksuit.schedules.base.schedule_base.ScheduleBase

A scheduler that switches between multiple schedules based on the step number.

Initialize the scheduler.

Parameters:

• schedule_configs (collections.abc.Sequence[SequentialStepScheduleConfig]) – A list of schedule configurations.

• **kwargs – Additional arguments to pass to the parent class.

schedule_configs

get_sequential_schedule_config(step)

Parameters:

step (int)

Return type:

SequentialStepScheduleConfig