├── .gitignore ├── LICENSE ├── README.md ├── cosine_annealing_warmup ├── __init__.py └── scheduler.py ├── requirements.txt ├── setup.py └── src ├── plot001.png └── plot002.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/README.md -------------------------------------------------------------------------------- /cosine_annealing_warmup/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/cosine_annealing_warmup/__init__.py -------------------------------------------------------------------------------- /cosine_annealing_warmup/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/cosine_annealing_warmup/scheduler.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.7 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/setup.py -------------------------------------------------------------------------------- /src/plot001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/src/plot001.png -------------------------------------------------------------------------------- /src/plot002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/katsura-jp/pytorch-cosine-annealing-with-warmup/HEAD/src/plot002.png --------------------------------------------------------------------------------