OptimCheckpoint.load()#

classmethod OptimCheckpoint.load(path)[source]#

Loads a trusted checkpoint file onto the current JAX device.

Pickle files can execute code: only load files from trusted sources. Runtime version and state compatibility are checked by fit() when resuming, so a checkpoint can be loaded for inspection independently.