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

set gradient_merge_cond persistable to false (#35578)

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