Commit f75a6b86 authored by Michael Case's avatar Michael Case Committed by TensorFlower Gardener
Browse files

MomentumOptimizer has two required args which causes error in optimize_loss. 

optimize_loss function tries to initialize Optimizer like...
opt = OPTIMIZER_CLS_NAMES[optimizer](learning_rate=lr)

This does not work for MomentumOptimizer which also has required arg
of 'momentum'.

PiperOrigin-RevId: 173415707
parent 805594a4
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment