/lookahead_keras

Keras Implementation of the Lookahead Optimizer

Primary LanguagePythonMIT LicenseMIT

Lookahead Optimizer

Reference

Michael R. Zhang, James Lucas, Geoffrey Hinton, Jimmy Ba. Lookahead Optimizer: k steps forward, 1 step back. [Arxiv]

Usage

from keras.optimizers import SGD
from lookahead import Lookahead

sgd = SGD(lr=0.001)
lookopt = Lookahead(optimizer=sgd, k=3, alpha=0.5)

model.compile(optimizer=lookopt, ...)