Skip to content

Latest commit

 

History

History
15 lines (11 loc) · 393 Bytes

README.md

File metadata and controls

15 lines (11 loc) · 393 Bytes

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, ...)