提交 3f87accf 编写于 作者: M Manjunath Kudlur 提交者: Gerrit Code Review

Merge "ADAM: Mark beta*_power variables as non-trainable."

......@@ -103,8 +103,8 @@ class AdamOptimizer(optimizer.Optimizer):
# variable.
if self._beta1_power is None:
with ops.device(var_list[0].device):
self._beta1_power = variables.Variable(self._beta1, name="beta1_power")
self._beta2_power = variables.Variable(self._beta2, name="beta2_power")
self._beta1_power = variables.Variable(self._beta1, name="beta1_power", trainable=False)
self._beta2_power = variables.Variable(self._beta2, name="beta2_power", trainable=False)
# Create slots for the first and second moments.
for v in var_list:
self._zeros_slot(v, "m", self._name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册