diff --git a/tools/train.py b/tools/train.py index dd2edbd4383524aa038347ea00997c997b980e93..b2632f4934eb21823e0b8b50a676404202456dd1 100644 --- a/tools/train.py +++ b/tools/train.py @@ -255,8 +255,9 @@ def main(): train_stats.update(stats) logs = train_stats.log() if it % cfg.log_iter == 0 and (not FLAGS.dist or trainer_id == 0): - strs = 'iter: {}, lr: {:.6f}, {}, time: {:.3f}, eta: {}'.format( - it, np.mean(outs[-1]), logs, time_cost, eta) + ips = float(cfg['TrainReader']['batch_size']) / time_cost + strs = 'iter: {}, lr: {:.6f}, {}, batch_cost: {:.5f} s, eta: {}, ips: {:.5f} images/sec'.format( + it, np.mean(outs[-1]), logs, time_cost, eta, ips) logger.info(strs) # NOTE : profiler tools, used for benchmark