From 00df73cf435271652cdb5e23fda11c1101db5b46 Mon Sep 17 00:00:00 2001 From: huaibovip <53330216+huaibovip@users.noreply.github.com> Date: Sat, 9 Sep 2023 20:54:57 +0800 Subject: [PATCH] [Fix] The keyword mode appears nested multiple times in the log (#1305) --- mmengine/runner/log_processor.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/mmengine/runner/log_processor.py b/mmengine/runner/log_processor.py index d3f9d957..c2254203 100644 --- a/mmengine/runner/log_processor.py +++ b/mmengine/runner/log_processor.py @@ -377,12 +377,11 @@ class LogProcessor: tag[key] = mode_history_scalars[key].current() # Update custom keys. for log_cfg in custom_cfg: - if not reserve_prefix: - data_src = log_cfg.pop('data_src') - log_name = f"{log_cfg.pop('log_name', data_src)}" - else: - data_src = f"{mode}/{log_cfg.pop('data_src')}" - log_name = f"{mode}/{log_cfg.pop('log_name', data_src)}" + data_src = log_cfg.pop('data_src') + log_name = log_cfg.pop('log_name', data_src) + if reserve_prefix: + data_src = f'{mode}/{data_src}' + log_name = f'{mode}/{log_name}' # log item in custom_cfg could only exist in train or val # mode. if data_src in mode_history_scalars: