From 093068e4ffdb0bb6ab0b95c45d9616835e930ce0 Mon Sep 17 00:00:00 2001
From: zhouhui <90194592+kitecats@users.noreply.github.com>
Date: Tue, 4 Apr 2023 00:17:42 +0800
Subject: [PATCH] [Enhancement] Align the evaluation result in log (#1034)
* align the evaluation result in log
* align the evaluation result in log
* align the evaluation result in log
* align the evaluation result in log
* fix test log_processor
---
mmengine/runner/log_processor.py | 1 +
tests/test_runner/test_log_processor.py | 8 ++++----
2 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/mmengine/runner/log_processor.py b/mmengine/runner/log_processor.py
index 34d59fd0..1259b5cc 100644
--- a/mmengine/runner/log_processor.py
+++ b/mmengine/runner/log_processor.py
@@ -307,6 +307,7 @@ class LogProcessor:
tag[key] = value
# Log other messages.
log_items = []
+ log_str += ' '
for name, val in chain(tag.items(), non_scalar_tag.items(),
time_tag.items()):
if isinstance(val, float):
diff --git a/tests/test_runner/test_log_processor.py b/tests/test_runner/test_log_processor.py
index c1fc833e..ac6ff91c 100644
--- a/tests/test_runner/test_log_processor.py
+++ b/tests/test_runner/test_log_processor.py
@@ -164,14 +164,14 @@ class TestLogProcessor:
'cm: \ntensor([1, 2, 3])\n data_time: 1.0000')
if by_epoch:
if mode == 'test':
- assert out == 'Epoch(test) [5/5] ' + expect_metric_str
+ assert out == 'Epoch(test) [5/5] ' + expect_metric_str
else:
- assert out == 'Epoch(val) [1][10/10] ' + expect_metric_str
+ assert out == 'Epoch(val) [1][10/10] ' + expect_metric_str
else:
if mode == 'test':
- assert out == 'Iter(test) [5/5] ' + expect_metric_str
+ assert out == 'Iter(test) [5/5] ' + expect_metric_str
else:
- assert out == 'Iter(val) [10/10] ' + expect_metric_str
+ assert out == 'Iter(val) [10/10] ' + expect_metric_str
def test_collect_scalars(self):
history_count = np.ones(100)
--
GitLab