From a76e404d9c3e130a762fdcdc735437452fc58ae4 Mon Sep 17 00:00:00 2001 From: littletomatodonkey <2120160898@bit.edu.cn> Date: Tue, 8 Dec 2020 17:05:00 +0800 Subject: [PATCH] fix time sta (#457) --- tools/static/program.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tools/static/program.py b/tools/static/program.py index 8741cb09..92829d93 100644 --- a/tools/static/program.py +++ b/tools/static/program.py @@ -429,16 +429,16 @@ def run(dataloader, feed=feed_dict, fetch_list=fetch_list) batch_time.update(time.time() - tic) - tic = time.time() for i, m in enumerate(metrics): metric_list[i].update(np.mean(m), batch_size) if mode == "train": metric_list[-1].update(lr_scheduler.get_lr()) + fetchs_str = ''.join([str(m.value) + ' ' - for m in metric_list] + [batch_time.value]) + 's' + for m in metric_list] + [batch_time.mean]) + 's' ips_info = " ips: {:.5f} images/sec.".format(batch_size / - batch_time.val) + batch_time.avg) fetchs_str += ips_info if lr_scheduler is not None: @@ -471,6 +471,8 @@ def run(dataloader, logger.coloring(step_str, "PURPLE"), logger.coloring(fetchs_str, 'OKGREEN'))) + tic = time.time() + end_str = ''.join([str(m.mean) + ' ' for m in metric_list] + [batch_time.total]) + 's' ips_info = "ips: {:.5f} images/sec.".format(batch_size * batch_time.count / -- GitLab