From c46b2bfa860cd02eeeb656865a3541c7be4e3211 Mon Sep 17 00:00:00 2001
From: JiabinYang <marsyang199376@gmail.com>
Date: Wed, 5 Dec 2018 09:02:07 +0000
Subject: [PATCH] add train_id and trainer_num to select reader

---
 fluid/PaddleRec/word2vec/infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fluid/PaddleRec/word2vec/infer.py b/fluid/PaddleRec/word2vec/infer.py
index 76e04b15..28111588 100644
--- a/fluid/PaddleRec/word2vec/infer.py
+++ b/fluid/PaddleRec/word2vec/infer.py
@@ -134,12 +134,12 @@ def infer_during_train(args):
     exe = fluid.Executor(fluid.CPUPlace())
     Scope = fluid.Scope()
     inference_prog()
+    solved_new = True
     while True:
         time.sleep(60)
         current_list = os.listdir(args.model_output_dir)
         # logger.info("current_list is : {}".format(current_list))
         # logger.info("model_file_list is : {}".format(model_file_list))
-        solved_new = True
         if set(model_file_list) == set(current_list):
             if solved_new:
                 solved_new = False
-- 
GitLab