提交 bf6219f2 编写于 作者: T Tao Luo

refine benchmark log

test=develop
上级 23d3ea61
...@@ -256,8 +256,8 @@ def main(): ...@@ -256,8 +256,8 @@ def main():
logs = train_stats.log() logs = train_stats.log()
if it % cfg.log_iter == 0 and (not FLAGS.dist or trainer_id == 0): if it % cfg.log_iter == 0 and (not FLAGS.dist or trainer_id == 0):
ips = float(cfg['TrainReader']['batch_size']) / time_cost ips = float(cfg['TrainReader']['batch_size']) / time_cost
strs = 'iter: {}, lr: {:.6f}, {}, batch_cost: {:.5f} s, eta: {}, ips: {:.5f} images/sec'.format( strs = 'iter: {}, lr: {:.6f}, {}, eta: {}, batch_cost: {:.5f} sec, ips: {:.5f} images/sec'.format(
it, np.mean(outs[-1]), logs, time_cost, eta, ips) it, np.mean(outs[-1]), logs, eta, time_cost, ips)
logger.info(strs) logger.info(strs)
# NOTE : profiler tools, used for benchmark # NOTE : profiler tools, used for benchmark
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册