Skip to main content

Module scheduler

Module scheduler 

Source
Expand description

Learning rate schedulers for controlling the optimization process.

This module provides various strategies to adjust the learning rate during training, such as cosine annealing with linear warmup, to improve model convergence and performance.

Structs§

ConstantLr
Constant Learning Rate Scheduler.
CosineAnnealingLr
Cosine Annealing Learning Rate Scheduler with Linear Warmup.

Enums§

Lr