From a491c19cf6c1f7126614e7f8889c037493fe52f6 Mon Sep 17 00:00:00 2001
From: chenguowei01 <chenguowei01@baidu.com>
Date: Thu, 30 Jul 2020 20:27:45 +0800
Subject: [PATCH] add tqdm

---
 dygraph/core/val.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/dygraph/core/val.py b/dygraph/core/val.py
index 2ec5f3b8..3245be5a 100644
--- a/dygraph/core/val.py
+++ b/dygraph/core/val.py
@@ -44,7 +44,8 @@ def evaluate(model,
             len(eval_dataset), total_steps))
     timer = Timer()
     timer.start()
-    for step, (im, im_info, label) in enumerate(eval_dataset):
+    for step, (im, im_info, label) in tqdm.tqdm(
+            enumerate(eval_dataset), total=total_steps):
         im = to_variable(im)
         pred, _ = model(im)
         pred = pred.numpy().astype('float32')
@@ -68,7 +69,7 @@ def evaluate(model,
 
         time_step = timer.elapsed_time()
         remain_step = total_steps - step - 1
-        logging.info(
+        logging.debug(
             "[EVAL] Epoch={}, Step={}/{}, iou={:4f}, sec/step={:.4f} | ETA {}".
             format(epoch_id, step + 1, total_steps, iou, time_step,
                    calculate_eta(remain_step, time_step)))
@@ -76,7 +77,7 @@ def evaluate(model,
 
     category_iou, miou = conf_mat.mean_iou()
     category_acc, macc = conf_mat.accuracy()
-    logging.info("[EVAL] #image={} acc={:.4f} IoU={:.4f}".format(
+    logging.info("[EVAL] #Images={} mAcc={:.4f} mIoU={:.4f}".format(
         len(eval_dataset), macc, miou))
     logging.info("[EVAL] Category IoU: " + str(category_iou))
     logging.info("[EVAL] Category Acc: " + str(category_acc))
-- 
GitLab