forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix program

This commit is contained in:
tink2123 2020-05-19 11:29:52 +08:00
parent 3ffb3879c6
commit d9757c7ec2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/program.py
View File

@ -269,7 +269,7 @@ def train_eval_det_run(config, exe, train_info_dict, eval_info_dict):
except fluid.core.EOFException: except fluid.core.EOFException:
train_loader.reset() 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_path = save_model_dir + "/iter_epoch_0"
save_model(train_info_dict['train_program'],save_path) save_model(train_info_dict['train_program'],save_path)
if epoch > 0 and epoch % save_epoch_step == 0: 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: except fluid.core.EOFException:
train_loader.reset() 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_path = save_model_dir + "/iter_epoch_0"
save_model(train_info_dict['train_program'],save_path) save_model(train_info_dict['train_program'],save_path)
if epoch > 0 and epoch % save_epoch_step == 0: if epoch > 0 and epoch % save_epoch_step == 0: