Tony-Y / pytorch_warmup

Learning Rate Warmup in PyTorch
https://tony-y.github.io/pytorch_warmup/
MIT License
392 stars 25 forks source link

Change the usage to respect "chaining" #6

Closed Tony-Y closed 2 years ago

Tony-Y commented 2 years ago

"chaining" is supported for PyTorch 1.4 or above: https://github.com/pytorch/pytorch/pull/26423