← all repositories

ildoonet/pytorch-gradual-warmup-lr

A PyTorch extension that gradually increases the learning rate during early training before switching to standard LR schedulers.

990 stars Python ML Frameworks
pytorch-gradual-warmup-lr
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

Provides a GradualWarmupScheduler for PyTorch that incrementally raises the learning rate over a specified number of epochs at the start of training, as proposed in the ‘Accurate, Large Minibatch SGD’ paper. It can be chained with other PyTorch schedulers such as StepLR or cosine annealing. This technique helps stabilize early training when using large batch sizes by preventing large gradient updates initially.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.