fix as comment

open-mmlab · Aug 25, 2022 · 231e1b0 · 231e1b0
1 parent c1ac287
commit 231e1b0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/mmengine/runner/log_processor.py b/mmengine/runner/log_processor.py
@@ -145,7 +145,7 @@ def get_log_after_iter(self, runner, batch_idx: int,
             # Epoch(train)  [9][100/270]
             max_dataloader_len = len(current_loop.dataloader)
             max_str_len = len(str(max_dataloader_len))
-            cur_iter_str = str(cur_iter).rjust(max_str_len, '0')
+            cur_iter_str = str(cur_iter).rjust(max_str_len)
 
             if mode in ['train', 'val']:
                 # Right Align the epoch log:
@@ -165,13 +165,13 @@ def get_log_after_iter(self, runner, batch_idx: int,
         else:
             if mode == 'train':
                 max_str_len = len(str(runner.max_iters))
-                cur_iter_str = str(cur_iter).rjust(max_str_len, '0')
+                cur_iter_str = str(cur_iter).rjust(max_str_len)
                 log_str = (f'Iter({mode}) '
                            f'[{cur_iter_str}/{runner.max_iters}]  ')
             else:
                 max_dataloader_len = len(current_loop.dataloader)
                 max_str_len = len(str(max_dataloader_len))
-                cur_iter_str = str(batch_idx + 1).rjust(max_str_len, '0')
+                cur_iter_str = str(batch_idx + 1).rjust(max_str_len)
                 log_str = (f'Iter({mode}) [{cur_iter_str}'
                            f'/{len(current_loop.dataloader)}]  ')
         # Concatenate lr, momentum string with log header.

diff --git a/tests/test_runner/test_log_processor.py b/tests/test_runner/test_log_processor.py
@@ -91,7 +91,7 @@ def test_get_log_after_iter(self, by_epoch, mode):
         if by_epoch:
             if mode in ['train', 'val']:
                 cur_epoch = log_processor._get_epoch(self.runner, mode)
-                log_str = (f'Epoch({mode})  [{cur_epoch}][02/'
+                log_str = (f'Epoch({mode})  [{cur_epoch}][ 2/'
                            f'{len(cur_loop.dataloader)}]  ')
             else:
                 log_str = (f'Epoch({mode}) [2/{len(cur_loop.dataloader)}]  ')
@@ -116,7 +116,7 @@ def test_get_log_after_iter(self, by_epoch, mode):
                 log_str = f'Iter({mode}) [11/{max_iters}]  '
             elif mode == 'val':
                 max_iters = len(cur_loop.dataloader)
-                log_str = f'Iter({mode}) [02/{max_iters}]  '
+                log_str = f'Iter({mode}) [ 2/{max_iters}]  '
             else:
                 max_iters = len(cur_loop.dataloader)
                 log_str = f'Iter({mode}) [2/{max_iters}]  '