pimm.export.load_checkpoint_metadata#

load_checkpoint_metadata(checkpoint_path: str, device: str | device = 'cpu', weights_only: bool = False) Dict[str, Any][source]#

Load checkpoint metadata (everything except state_dict).

Returns metadata dict with keys like ‘epoch’, ‘config’, ‘args’, etc.