未验证 提交 6f2a096b 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者: GitHub

fix ips info (#1306)

上级 46f86bc8
......@@ -428,9 +428,8 @@ def run(dataloader,
end_str = ' '.join([str(m.mean) for m in metric_dict.values()] +
[metric_dict["batch_time"].total])
ips_info = "ips: {:.5f} images/sec.".format(
batch_size * metric_dict["batch_time"].count /
metric_dict["batch_time"].sum)
ips_info = "ips: {:.5f} images/sec.".format(batch_size /
metric_dict["batch_time"].avg)
if mode == 'eval':
logger.info("END {:s} {:s} {:s}".format(mode, end_str, ips_info))
else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册