Merge pull request #9901 from Yancey1989/fix_duplicate_lr_op

Fix duplicate lr op after distribute transpiler
7 years ago · 9bc0c23b17
parent c1bf06f9d9 253441b553
commit 9bc0c23b17
1 changed files with 2 additions and 0 deletions
--- a/python/paddle/fluid/distribute_transpiler.py
+++ b/python/paddle/fluid/distribute_transpiler.py
@ -1115,4 +1115,6 @@ class DistributeTranspiler:
            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