fix program

d9757c7e · tink2123 · 3ffb3879 · d9757c7e
Commit d9757c7e authored May 19, 2020 by tink2123
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/program.py tools/program.py +2 -2

No files found.
--- 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: