From e2841359996b87b080ff30db862ef63c81d25790 Mon Sep 17 00:00:00 2001
From: buchongyu <18001307871@163.com>
Date: Mon, 11 Jul 2022 11:39:00 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=9D=99=E6=80=81=E5=9B=BE?=
 =?UTF-8?q?=E5=A4=9A=E6=9C=BA=E5=A4=9A=E5=8D=A1=E6=97=A0=E6=B3=95=E8=AE=AD?=
 =?UTF-8?q?=E7=BB=83bug=20(#6390)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 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