Skip to content

P
PaddleOCR

PaddlePaddle / PaddleOCR
大约 1 年 前同步成功

通知 1528
Star 32962
Fork 6643
P
PaddleOCR
提交 d9757c7e 编写于 作者: T tink2123
浏览文件
操作

fix program

上级 3ffb3879
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
tools/program.py
浏览文件 @ d9757c7e
...@@ -269,7 +269,7 @@ def train_eval_det_run(config, exe, train_info_dict, eval_info_dict): ...@@ -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): ...@@ -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:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或