[Improve] improve checkpoint loading log (#1446)

open-mmlab · Nov 2, 2021 · 2fe7d6c · 2fe7d6c
1 parent f9be3bc
commit 2fe7d6c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/mmcv/runner/base_runner.py b/mmcv/runner/base_runner.py
@@ -334,8 +334,6 @@ def load_checkpoint(self,
                         map_location='cpu',
                         strict=False,
                         revise_keys=[(r'^module.', '')]):
-
-        self.logger.info('load checkpoint from %s', filename)
         return load_checkpoint(
             self.model,
             filename,

diff --git a/mmcv/runner/checkpoint.py b/mmcv/runner/checkpoint.py
@@ -240,7 +240,8 @@ def load_checkpoint(cls, filename, map_location=None, logger=None):
 
         checkpoint_loader = cls._get_checkpoint_loader(filename)
         class_name = checkpoint_loader.__name__
-        mmcv.print_log(f'Use {class_name} loader', logger)
+        mmcv.print_log(
+            f'load checkpoint from {class_name[10:]} path: {filename}', logger)
         return checkpoint_loader(filename, map_location)