load_checkpoint_state_dict#

load_checkpoint_state_dict(ckpt_path)#

Load a .ckpt file and return a flat state_dict-like mapping.

Supports both Lightning checkpoints (with a ‘state_dict’ key) and plain torch.save(state_dict) style checkpoints.