Skip to content

Commit 5da4ece

Browse files
authored
Merge pull request #13630 from typhoonzero/pick_trainer_program_fix
fix graph num
2 parents d23c3ff + 71035e9 commit 5da4ece

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)