mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
* Format log of

* Fix as comment

* Update mmengine/runner/checkpoint.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
This commit is contained in:
Mashiro 2022-11-15 20:18:33 +08:00 committed by GitHub
parent 48650f3bac
commit 6eced86f6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mmengine/runner/checkpoint.py
View File

@ -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:
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)