kd.ckpts

[[Source]]

Checkpoints API.

Symbols

Module

kd.ckpts.items	Checkpoint handler.

Class

kd.ckpts.AbstractPartialLoader	Abstract class for partial checkpoint loaders.
kd.ckpts.Checkpointer	Wrapper around Orbax CheckpointManager.
kd.ckpts.MultiTransform	Transform which applies multiple transformations sequentially.
kd.ckpts.NoopCheckpointer	Does nothing.
kd.ckpts.NoopTransform	init_transform that does nothing.
kd.ckpts.PartialKauldronLoader	Partial loader for Kauldron checkpoints.

Function

kd.ckpts.workdir_from_xid