From 2fe7d6c3ec1c86dc8e7131bde4a869bf30361f07 Mon Sep 17 00:00:00 2001 From: WangJiaZhen <47851024+teamwong111@users.noreply.github.com> Date: Tue, 2 Nov 2021 20:52:23 +0800 Subject: [PATCH] [Improve] improve checkpoint loading log (#1446) --- mmcv/runner/base_runner.py | 2 -- mmcv/runner/checkpoint.py | 3 ++- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/mmcv/runner/base_runner.py b/mmcv/runner/base_runner.py index d7ed4cb713..25cd98f51c 100644 --- 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 index 4db75d23f7..6ad605b854 100644 --- 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)