This commit is contained in:
LDOUBLEV 2020-07-07 02:51:50 +00:00
parent e222bc9e64
commit f5e0e9d050
1 changed files with 2 additions and 2 deletions

View File

@ -253,7 +253,7 @@ def train_eval_det_run(config, exe, train_info_dict, eval_info_dict):
t2 = time.time() t2 = time.time()
train_batch_elapse = t2 - t1 train_batch_elapse = t2 - t1
train_stats.update(stats) train_stats.update(stats)
if train_batch_id > start_eval_step and train_batch_id \ if train_batch_id > start_eval_step and (train_batch_id -start_eval_step) \
% print_batch_step == 0: % print_batch_step == 0:
logs = train_stats.log() logs = train_stats.log()
strs = 'epoch: {}, iter: {}, {}, time: {:.3f}'.format( strs = 'epoch: {}, iter: {}, {}, time: {:.3f}'.format(
@ -338,7 +338,7 @@ def train_eval_rec_run(config, exe, train_info_dict, eval_info_dict):
train_batch_elapse = t2 - t1 train_batch_elapse = t2 - t1
stats = {'loss': loss, 'acc': acc} stats = {'loss': loss, 'acc': acc}
train_stats.update(stats) train_stats.update(stats)
if train_batch_id > start_eval_step and train_batch_id \ if train_batch_id > start_eval_step and (train_batch_id - start_eval_step) \
% print_batch_step == 0: % print_batch_step == 0:
logs = train_stats.log() logs = train_stats.log()
strs = 'epoch: {}, iter: {}, lr: {:.6f}, {}, time: {:.3f}'.format( strs = 'epoch: {}, iter: {}, lr: {:.6f}, {}, time: {:.3f}'.format(