未验证 提交 ca879e5a 编写于 作者: X xujiaqi01 提交者: GitHub

fix skip_op bug (#21418)

test=develop
上级 b1627455
......@@ -184,7 +184,7 @@ class DownpourSGD(DeviceWorker):
dense_table.dense_value_name.extend(i.dense_variable_name)
dense_table.dense_grad_name.extend(
i.dense_gradient_variable_name)
downpour.skip_ops.extend(worker.get_desc().skip_op)
downpour.skip_ops.extend(worker.get_desc().skip_op)
if self._infer:
downpour.push_dense = False
downpour.push_sparse = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册