From 5b648c119fc0f8789b6cb285d00e8c7fe8cba2ba Mon Sep 17 00:00:00 2001
From: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Date: Tue, 19 Jul 2022 18:30:59 +0800
Subject: [PATCH] [Fix] Fix LogProcessor._get_iter (#370)
---
mmengine/logging/log_processor.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmengine/logging/log_processor.py b/mmengine/logging/log_processor.py
index d1fd8a4b..ee2e047c 100644
--- a/mmengine/logging/log_processor.py
+++ b/mmengine/logging/log_processor.py
@@ -369,7 +369,7 @@ class LogProcessor:
Returns:
int: The current global iter or inner iter.
"""
- if self.by_epoch and batch_idx:
+ if self.by_epoch and batch_idx is not None:
current_iter = batch_idx + 1
else:
current_iter = runner.iter + 1
--
GitLab