Commit c9d464c3 authored by WangJiaZhen's avatar WangJiaZhen Committed by Wenwei Zhang
Browse files

[Improve] improve checkpoint loading log (#1446)

parent fd6eab69
...@@ -334,8 +334,6 @@ class BaseRunner(metaclass=ABCMeta): ...@@ -334,8 +334,6 @@ class BaseRunner(metaclass=ABCMeta):
map_location='cpu', map_location='cpu',
strict=False, strict=False,
revise_keys=[(r'^module.', '')]): revise_keys=[(r'^module.', '')]):
self.logger.info('load checkpoint from %s', filename)
return load_checkpoint( return load_checkpoint(
self.model, self.model,
filename, filename,
......
...@@ -240,7 +240,8 @@ class CheckpointLoader: ...@@ -240,7 +240,8 @@ class CheckpointLoader:
checkpoint_loader = cls._get_checkpoint_loader(filename) checkpoint_loader = cls._get_checkpoint_loader(filename)
class_name = checkpoint_loader.__name__ 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) return checkpoint_loader(filename, map_location)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment