Skip to content
Snippets Groups Projects
Unverified Commit 6eced86f authored by Mashiro's avatar Mashiro Committed by GitHub
Browse files

[Enhancement] Format the log of `CheckpointLoader.load_checkpoint` (#685)


* Format log of

* Fix as comment

* Update mmengine/runner/checkpoint.py

Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>

Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
parent 48650f3b
No related branches found
No related tags found
No related merge requests found
...@@ -236,15 +236,14 @@ class CheckpointLoader: ...@@ -236,15 +236,14 @@ class CheckpointLoader:
return cls._schemes[p] return cls._schemes[p]
@classmethod @classmethod
def load_checkpoint(cls, filename, map_location=None, logger=None): def load_checkpoint(cls, filename, map_location=None, logger='current'):
"""load checkpoint through URL scheme path. """load checkpoint through URL scheme path.
Args: Args:
filename (str): checkpoint file name with given prefix filename (str): checkpoint file name with given prefix
map_location (str, optional): Same as :func:`torch.load`. map_location (str, optional): Same as :func:`torch.load`.
Default: None Default: None
logger (:mod:`logging.Logger`, optional): The logger for message. logger (str): The logger for message. Defaults to 'current'.
Default: None
Returns: Returns:
dict or OrderedDict: The loaded checkpoint. dict or OrderedDict: The loaded checkpoint.
...@@ -252,8 +251,10 @@ class CheckpointLoader: ...@@ -252,8 +251,10 @@ 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__
print_log(f'{class_name[10:]} loads checkpoint from path: {filename}', print_log(
logger) f'Loads checkpoint by {class_name[10:]} backend from path: '
f'{filename}',
logger=logger)
return checkpoint_loader(filename, map_location) return checkpoint_loader(filename, map_location)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment