class_docstr: Interface to save/load checkpoints as they are saved through the ``Strategy``.
  
  .. warning::  This is an :ref:`experimental <versioning:Experimental API>` feature.
  
  Typically most plugins either use the Torch based IO Plugin; ``TorchCheckpointIO`` but may
  require particular handling depending on the plugin.
  
  In addition, you can pass a custom ``CheckpointIO`` by extending this class and passing it
  to the Trainer, i.e ``Trainer(plugins=[MyCustomCheckpointIO()])``.
  
  .. note::
  
      For some plugins, it is not possible to use a custom checkpoint plugin as checkpointing logic is not
      modifiable.
docstr: Load checkpoint from a path when resuming or loading ckpt for test/validate/predict stages.
  
  Args:
      path: Path to checkpoint
      map_location: a function, :class:`torch.device`, string or a dict specifying how to remap storage
          locations.
  
  Returns: The loaded checkpoint.
docstr_tok: ["Load","checkpoint","from","a","path","when","resuming","or","loading","ckpt","for","test\/validate\/predict","stages",".","Args",":","path",":","Path","to","checkpoint","map_location",":","a","function",",",":","class",":","`","torch.device","`",",","string","or","a","dict","specifying","how","to","remap","storage","locations",".","Returns",":","The","loaded","checkpoint","."]
code: def load_checkpoint(
      self, path: _PATH, map_location: Optional[Any] = None
  ) -> Dict[str, Any]:
      """Load checkpoint from a path when resuming or loading ckpt for test/validate/predict stages.
  
      Args:
          path: Path to checkpoint
          map_location: a function, :class:`torch.device`, string or a dict specifying how to remap storage
              locations.
  
      Returns: The loaded checkpoint.
  
      """
  
file_path: pytorch-lightning/src/lightning/fabric/plugins/io/checkpoint_io.py
imports: from abc import ABC, abstractmethod
  from typing import Any, Dict, Optional
  from lightning.fabric.utilities.types import _PATH
label_desc: Use image node_id 2 for calling the CheckpointIO obj's underlying member method code with example usage: obj.load_checkpoint(path, map_location) without return types