未验证 提交 12eff933 编写于 作者: K kangguangli 提交者: GitHub

remove with_data_parallel in fleet (#51031)

上级 efd8346f
...@@ -1453,7 +1453,7 @@ class Fleet: ...@@ -1453,7 +1453,7 @@ class Fleet:
compiled_program = compiler.CompiledProgram( compiled_program = compiler.CompiledProgram(
self.origin_main_program self.origin_main_program
).with_data_parallel(loss_name=loss.name, share_vars_from=None) )
loss.block.program._graph = compiled_program loss.block.program._graph = compiled_program
return self.user_defined_optimizer.minimize( return self.user_defined_optimizer.minimize(
loss, loss,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册