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):
map_location='cpu',
strict=False,
revise_keys=[(r'^module.', '')]):
self.logger.info('load checkpoint from %s', filename)
return load_checkpoint(
self.model,
filename,
......
......@@ -240,7 +240,8 @@ class CheckpointLoader:
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)
......
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