Commit 6fde1c56 authored by Zhenyu Tan's avatar Zhenyu Tan Committed by TensorFlower Gardener
Browse files

Fix NAdam optimizer, 1) assert dependency between iteration update and variable 

update, 2) include momentum caching as state variable.

PiperOrigin-RevId: 226429048
parent 6e5f37d5
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