/pytorch_warmup-scheduler

A PyTorch implementation of the warmup scheduler

Primary LanguageJupyter NotebookMIT LicenseMIT

PyTorch Warmup Scheduler

PyPI version Open In Colab

This is a PyTorch implementation of the warmup scheduler.

Installation

pip install pytorch-warmup-scheduler

References

  • Goyal, Priya, Piotr Dollár, Ross Girshick, Pieter Noordhuis, Lukasz Wesolowski, Aapo Kyrola, Andrew Tulloch, Yangqing Jia, and Kaiming He. "Accurate, Large Minibatch SGD: Training ImageNet in 1 Hour." arXiv preprint arXiv:1706.02677 (2017). arXiv:1706.02677