Merge pull request #13619 from typhoonzero/fix_graph_num

fix graph num

Merge pull request #13619 from typhoonzero/fix_graph_num
fix graph num
18e5dccc · Wu Yi · GitHub · 9e8d372f · 5b152b1f · 18e5dccc
显示空白变更内容
内联并排

Showing with 3 addition and 0 deletion

python/paddle/fluid/transpiler/distribute_transpiler.py python/paddle/fluid/transpiler/distribute_transpiler.py +3 -0

未找到文件。
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@@ -470,7 +470,10 @@ class DistributeTranspiler(object):
        """
        # remove optimize ops and add a send op to main_program
        # FIXME(typhoonzero): Also ops like clip_gradient, lrn_decay?
+        lr_ops = self._get_lr_ops()
        delete_ops(self.origin_program.global_block(), self.optimize_ops)
+        delete_ops(self.origin_program.global_block(), lr_ops)
+
        self.origin_program.__str__()

        if wait_port: