未验证 提交 47d15a30 编写于 作者: Z Zhong Hui 提交者: GitHub

set gradient_merge_cond persistable to false (#35578)

上级 98d047d7
...@@ -6918,7 +6918,7 @@ class GradientMergeOptimizer(object): ...@@ -6918,7 +6918,7 @@ class GradientMergeOptimizer(object):
shape=[1], shape=[1],
value=bool(0), value=bool(0),
dtype='bool', dtype='bool',
persistable=True, persistable=False,
force_cpu=True) force_cpu=True)
with device_guard("cpu"): with device_guard("cpu"):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册