Skip to content

Commit 71035e9

Browse files
committed
fix graph num
1 parent 4aed00e commit 71035e9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -470,7 +470,10 @@ def get_trainer_program(self, wait_port=True):
470470
"""
471471
# remove optimize ops and add a send op to main_program
472472
# FIXME(typhoonzero): Also ops like clip_gradient, lrn_decay?
473+
lr_ops = self._get_lr_ops()
473474
delete_ops(self.origin_program.global_block(), self.optimize_ops)
475+
delete_ops(self.origin_program.global_block(), lr_ops)
476+
474477
self.origin_program.__str__()
475478

476479
if wait_port:

0 commit comments

Comments
 (0)