Merge pull request #9901 from Yancey1989/fix_duplicate_lr_op

Fix duplicate lr op after distribute transpiler
PaddlePaddle · Apr 13, 2018 · 9bc0c23 · 9bc0c23
2 parents c1bf06f + 253441b
commit 9bc0c23
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/paddle/fluid/distribute_transpiler.py b/python/paddle/fluid/distribute_transpiler.py
@@ -1115,4 +1115,6 @@ def _get_lr_ops(self):
             for op2 in find_ops:
                 if ufind.is_connected(op1, op2):
                     lr_ops.append(op1)
+                    # we only need to append op for once
+                    break
         return lr_ops