CheckpointLoader¶
- class mmengine.runner.CheckpointLoader[源代码]¶
A general checkpoint loader to manage all schemes.
- classmethod load_checkpoint(filename, map_location=None, logger='current')[源代码]¶
Load checkpoint through URL scheme path.
- 参数:
filename (str) – checkpoint file name with given prefix
map_location (str, optional) – Same as
torch.load()
. Defaults to Nonelogger (str) – The logger for message. Defaults to ‘current’.
- 返回:
The loaded checkpoint.
- 返回类型:
dict or OrderedDict
- classmethod register_scheme(prefixes, loader=None, force=False)[源代码]¶
Register a loader to CheckpointLoader.
This method can be used as a normal class method or a decorator.
- 参数:
loader. (The prefix of the registered) –
loader (function, optional) – The loader function to be registered. When this method is used as a decorator, loader is None. Defaults to None.
force (bool, optional) – Whether to override the loader if the prefix has already been registered. Defaults to False.