From d9757c7ec29305da41b7ded8ee139c3b6cb12042 Mon Sep 17 00:00:00 2001 From: tink2123 Date: Tue, 19 May 2020 11:29:52 +0800 Subject: [PATCH] fix program --- tools/program.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/program.py b/tools/program.py index f5572073..a12e3360 100755 --- a/tools/program.py +++ b/tools/program.py @@ -269,7 +269,7 @@ def train_eval_det_run(config, exe, train_info_dict, eval_info_dict): except fluid.core.EOFException: train_loader.reset() - if save_epoch_step == 1: + if epoch == 0 and save_epoch_step == 1: save_path = save_model_dir + "/iter_epoch_0" save_model(train_info_dict['train_program'],save_path) if epoch > 0 and epoch % save_epoch_step == 0: @@ -348,7 +348,7 @@ def train_eval_rec_run(config, exe, train_info_dict, eval_info_dict): except fluid.core.EOFException: train_loader.reset() - if save_epoch_step == 1: + if epoch ==0 and save_epoch_step == 1: save_path = save_model_dir + "/iter_epoch_0" save_model(train_info_dict['train_program'],save_path) if epoch > 0 and epoch % save_epoch_step == 0: -- GitLab