未验证 提交 23d3ea61 编写于 作者: T Tao Luo 提交者: GitHub

add ips for detection models (#1507)

test=develop
上级 5549e083
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册