diff --git a/mmengine/runner/log_processor.py b/mmengine/runner/log_processor.py index 41fc3bc546..874a1ec127 100644 --- a/mmengine/runner/log_processor.py +++ b/mmengine/runner/log_processor.py @@ -197,7 +197,9 @@ def get_log_after_iter(self, runner, batch_idx: int, # If cuda is available, the max memory occupied should be calculated. if is_cuda_available(): - log_str += f'memory: {self._get_max_memory(runner)} ' + max_memory = self._get_max_memory(runner) + log_str += f'memory: {max_memory} ' + tag['memory'] = max_memory # Loop left keys to fill `log_str`. if mode in ('train', 'val'): log_items = []