stable_ssl.schedulers

LinearWarmup(optimizer, total_steps[, ...])	Create a linear warmup scheduler.
LinearWarmupCosineAnnealing(optimizer, ...)	Combine linear warmup with cosine annealing decay.
LinearWarmupCyclicAnnealing(optimizer, ...)	Combine linear warmup with cyclic cosine annealing.
LinearWarmupThreeStepsAnnealing(optimizer, ...)	Combine linear warmup with a three-step learning rate annealing.