From bf6219f2c8e9434a33003843b0dedf5ea73b40b6 Mon Sep 17 00:00:00 2001
From: Tao Luo <luotao02@baidu.com>
Date: Fri, 25 Sep 2020 18:48:16 +0800
Subject: [PATCH] refine benchmark log

test=develop
---
 tools/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index b2632f493..0004bea06 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -256,8 +256,8 @@ def main():
         logs = train_stats.log()
         if it % cfg.log_iter == 0 and (not FLAGS.dist or trainer_id == 0):
             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)
+            strs = 'iter: {}, lr: {:.6f}, {}, eta: {}, batch_cost: {:.5f} sec, ips: {:.5f} images/sec'.format(
+                it, np.mean(outs[-1]), logs, eta, time_cost, ips)
             logger.info(strs)
 
         # NOTE : profiler tools, used for benchmark
-- 
GitLab