From eb2512993586143c9b900fb7a55d46c94359cec9 Mon Sep 17 00:00:00 2001
From: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Date: Tue, 2 Aug 2022 19:01:06 +0800
Subject: [PATCH] [Feature]: add epoch log scalar (#391)
---
mmengine/logging/log_processor.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/mmengine/logging/log_processor.py b/mmengine/logging/log_processor.py
index ee2e047c..3811d714 100644
--- a/mmengine/logging/log_processor.py
+++ b/mmengine/logging/log_processor.py
@@ -143,6 +143,7 @@ class LogProcessor:
if self.by_epoch:
if mode in ['train', 'val']:
cur_epoch = self._get_epoch(runner, mode)
+ tag['epoch'] = cur_epoch
log_str = (f'Epoch({mode}) [{cur_epoch}]'
f'[{cur_iter}/{len(current_loop.dataloader)}] ')
else:
--
GitLab