jettify / pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch
Apache License 2.0
3.04k stars 300 forks source link

add swats #151

Closed sailfish009 closed 4 years ago

sailfish009 commented 4 years ago

Adam_SGD:

"SWATS from Keskar & Socher (2017) a high-scoring paper by ICLR in 2018, a method proposed to automatically switch from Adam to SGD for better generalization performance. The idea of the algorithm itself is very simple. It uses Adam, which works well despite minimal tuning, but after learning until a certain stage, it is taken over by SGD."

https://github.com/Mrpatekful/swats

jettify commented 4 years ago

FYI https://github.com/jettify/pytorch-optimizer/pull/178