mixmo.core.scheduler

Scheduler definitions and factory

Functions

get_scheduler(lr_schedule, optimizer, …)	Build the scheduler object
get_warmup_scheduler(optimizer, warmup_period)	Build a Scheduler instance with warmup

Classes

GradualWarmupScheduler(optimizer, …)	Gradually warm-up(increasing) learning rate in optimizer.
MultiGammaStepLR(optimizer, …[, last_epoch])	Multi step decay scheduler, with decay applied to the learning rate every set milestone