From ccb32c7e31577f145628ec1d7a6245443ea7d5ac Mon Sep 17 00:00:00 2001
From: Yanyi Liu <wolfsonliu@163.com>
Date: Sat, 8 Oct 2022 19:53:05 +0800
Subject: [PATCH] [Fix] fix loss smooth when loss name doesn't start with
`loss` (#539)
* fix loss mean
* update
---
mmengine/runner/log_processor.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/mmengine/runner/log_processor.py b/mmengine/runner/log_processor.py
index 756ea00f..b666fddc 100644
--- a/mmengine/runner/log_processor.py
+++ b/mmengine/runner/log_processor.py
@@ -153,7 +153,7 @@ class LogProcessor:
log_str = (f'Iter({mode}) '
f'[{cur_iter}/{runner.max_iters}] ')
else:
- log_str = (f'Iter({mode}) [{batch_idx+1}'
+ log_str = (f'Iter({mode}) [{batch_idx + 1}'
f'/{len(current_loop.dataloader)}] ')
# Concatenate lr, momentum string with log header.
log_str += f'{lr_str} '
@@ -266,8 +266,7 @@ class LogProcessor:
mode_history_scalars[key] = log_buffer
for key in mode_history_scalars:
# Update the latest learning rate and smoothed time logs.
- if key.startswith('loss') or key in ('time', 'data_time',
- 'grad_norm'):
+ if 'loss' in key or key in ('time', 'data_time', 'grad_norm'):
tag[key] = mode_history_scalars[key].mean(self.window_size)
else:
# Default statistic method is current.
--
GitLab