core.trainers.mixins.scheduler

core.trainers.mixins.scheduler

Module for Axolotl trainer scheduler mixin

Classes

Name	Description
SchedulerMixin	Mixin class for scheduler setup in CausalTrainer.

core.trainers.mixins.scheduler.SchedulerMixin()

Mixin class for scheduler setup in CausalTrainer.

Name	Description
create_scheduler	Set up the scheduler. The optimizer of the trainer must have been set up either before this method is called or

core.trainers.mixins.scheduler.SchedulerMixin.create_scheduler(
    num_training_steps,
    optimizer=None,
)

Set up the scheduler. The optimizer of the trainer must have been set up either before this method is called or passed as an argument.

Name	Type	Description	Default
num_training_steps	int	The number of training steps to do.	required
optimizer	torch.optim.Optimizer	The training optimizer	`None`