neuralbench.utils.load_checkpoint¶

neuralbench.utils.load_checkpoint(brain_model: Module, checkpoint_path: str | Path, logger: Logger) → Module[source][source]¶: Load checkpoint through state_dicts.

Note

While pytorch-lightning exposes ways to do this, we implement checkpoint loading directly for more fine-grained control.