From eed41762b7302cdab090ba0c403a493b70a05043 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 14 Jul 2020 14:52:12 +0800 Subject: [PATCH] fix bug det train --- tools/program.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/program.py b/tools/program.py index 6a51e5c3..ff8743f1 100755 --- a/tools/program.py +++ b/tools/program.py @@ -256,15 +256,15 @@ def train_eval_det_run(config, exe, train_info_dict, eval_info_dict): t2 = time.time() train_batch_elapse = t2 - t1 train_stats.update(stats) - if train_batch_id > start_eval_step and (train_batch_id -start_eval_step) \ + if train_batch_id > 0 and train_batch_id \ % print_batch_step == 0: logs = train_stats.log() strs = 'epoch: {}, iter: {}, {}, time: {:.3f}'.format( epoch, train_batch_id, logs, train_batch_elapse) logger.info(strs) - if train_batch_id > 0 and\ - train_batch_id % eval_batch_step == 0: + if train_batch_id > start_eval_step and\ + (train_batch_id - start_eval_step) % eval_batch_step == 0: metrics = eval_det_run(exe, config, eval_info_dict, "eval") hmean = metrics['hmean'] if hmean >= best_eval_hmean: -- GitLab