Formatting fixes and internal changes.

PiperOrigin-RevId: 414656804
This commit is contained in:
Loren Maggiore
2021-12-07 09:58:30 +00:00
committed by Saran Tunyasuvunakool
parent 826ff89f21
commit d55816beb3
6 changed files with 83 additions and 57 deletions
+1 -1
View File
@@ -132,7 +132,7 @@ class Optimizer(object):
elif '_states' in self.__dict__ and name in self._states:
return self._states[name]
else:
object.__getattr__(self, name)
object.__getattribute__(self, name)
def step(self, params, grads, states, itr=None):
"""Takes a single optimizer step.