Shortcuts

CheckpointLoader

class mmengine.runner.CheckpointLoader[source]

A general checkpoint loader to manage all schemes.

classmethod load_checkpoint(filename, map_location=None, logger='current')[source]

Load checkpoint through URL scheme path.

Parameters:
  • filename (str) – checkpoint file name with given prefix

  • map_location (str, optional) – Same as torch.load(). Defaults to None

  • logger (str) – The logger for message. Defaults to ‘current’.

Returns:

The loaded checkpoint.

Return type:

dict or OrderedDict

classmethod register_scheme(prefixes, loader=None, force=False)[source]

Register a loader to CheckpointLoader.

This method can be used as a normal class method or a decorator.

Parameters:
  • prefixes (str or list[str] or tuple[str]) –

  • 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.