From c3faafbb24c4b95981c880905949252247514830 Mon Sep 17 00:00:00 2001
From: buchongyu <18001307871@163.com>
Date: Mon, 11 Jul 2022 15:30:55 +0800
Subject: [PATCH] [release/2.4] fix train_multi_machine bug in static ppdet
 (#6398)

---
 static/tools/train_multi_machine.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/static/tools/train_multi_machine.py b/static/tools/train_multi_machine.py
index 098845d0a..d4624b6ee 100644
--- a/static/tools/train_multi_machine.py
+++ b/static/tools/train_multi_machine.py
@@ -192,7 +192,6 @@ def main():
                                                          extra_keys)
 
     exe.run(startup_prog)
-    compiled_train_prog = fleet.main_program
 
     if FLAGS.eval:
         compiled_eval_prog = fluid.CompiledProgram(eval_prog)
@@ -253,7 +252,7 @@ def main():
         time_cost = np.mean(time_stat)
         eta_sec = (cfg.max_iters - it) * time_cost
         eta = str(datetime.timedelta(seconds=int(eta_sec)))
-        outs = exe.run(compiled_train_prog, fetch_list=train_values)
+        outs = exe.run(train_prog, fetch_list=train_values)
         stats = {k: np.array(v).mean() for k, v in zip(train_keys, outs[:-1])}
 
         # use vdl-paddle to log loss
-- 
GitLab