mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] The keyword mode appears nested multiple times in the log (#1305)

Browse Source
This commit is contained in:
huaibovip 2023-09-09 20:54:57 +08:00 committed by GitHub
parent 9a19f8bc89
commit 00df73cf43
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mmengine/runner/log_processor.py
View File

@ -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: