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

fix skip_op bug (#21418)

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