diff --git a/mmengine/logging/log_processor.py b/mmengine/logging/log_processor.py index ee2e047c8344a246333fd1cc55b2138f677c1ec0..3811d714d7e8868e27ee8d2bcb852acc5ab2d517 100644 --- a/mmengine/logging/log_processor.py +++ b/mmengine/logging/log_processor.py @@ -143,6 +143,7 @@ class LogProcessor: if self.by_epoch: if mode in ['train', 'val']: cur_epoch = self._get_epoch(runner, mode) + tag['epoch'] = cur_epoch log_str = (f'Epoch({mode}) [{cur_epoch}]' f'[{cur_iter}/{len(current_loop.dataloader)}] ') else: