From 60c2106f13d2ae0518cb4e19cd47f78b179036b3 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Wed, 16 Jun 2021 21:12:47 +0800
Subject: [PATCH] add feature calc log (#865)

---
 ppcls/engine/trainer.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py
index 4234d6294..b70793f3b 100644
--- a/ppcls/engine/trainer.py
+++ b/ppcls/engine/trainer.py
@@ -476,6 +476,10 @@ class Trainer(object):
         has_unique_id = False
         for idx, batch in enumerate(dataloader(
         )):  # load is very time-consuming
+            if idx % self.config["Global"]["print_batch_step"] == 0:
+                logger.info(
+                    f"{name} feature calculation process: [{idx}/{len(dataloader)}]"
+                )
             batch = [paddle.to_tensor(x) for x in batch]
             batch[1] = batch[1].reshape([-1, 1]).astype("int64")
             if len(batch) == 3: