rzhu3 / YellowFin_MXNet

auto-tuning momentum SGD optimizer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

YellowFin

YellowFin is an auto-tuning optimizer based on momentum SGD which requires no manual specification of learning rate and momentum. It measures the objective landscape on-the-fly and tune momentum as well as learning rate using local quadratic approximation.

The implementation here can be a drop-in replacement for any optimizer in MXNet (So far we only implemented and tested upon SGD and other optimizers are in the to-do list).

For more technical details, please refer to the paper YellowFin and the Art of Momentum Tuning.

About

auto-tuning momentum SGD optimizer

License:Apache License 2.0


Languages

Language:Python 100.0%