提交 c46b2bfa 编写于 作者: J JiabinYang

add train_id and trainer_num to select reader

上级 73992ce3
...@@ -134,12 +134,12 @@ def infer_during_train(args): ...@@ -134,12 +134,12 @@ def infer_during_train(args):
exe = fluid.Executor(fluid.CPUPlace()) exe = fluid.Executor(fluid.CPUPlace())
Scope = fluid.Scope() Scope = fluid.Scope()
inference_prog() inference_prog()
solved_new = True
while True: while True:
time.sleep(60) time.sleep(60)
current_list = os.listdir(args.model_output_dir) current_list = os.listdir(args.model_output_dir)
# logger.info("current_list is : {}".format(current_list)) # logger.info("current_list is : {}".format(current_list))
# logger.info("model_file_list is : {}".format(model_file_list)) # logger.info("model_file_list is : {}".format(model_file_list))
solved_new = True
if set(model_file_list) == set(current_list): if set(model_file_list) == set(current_list):
if solved_new: if solved_new:
solved_new = False solved_new = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册