fix duplicate lr op after distribute transpiler

7 years ago · 253441b553
parent 41a9146de3
commit 253441b553
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