gm.ckpts

[[Source]]

Checkpoints API.

Symbols

Class

gm.ckpts.AnchoredPolicyLoader	Loader for gm.nn.AnchoredPolicy models.
gm.ckpts.CheckpointPath	Hardcoded paths to Gemma checkpoints.
gm.ckpts.LoadCheckpoint	Loads weights from a Gemma checkpoint.
gm.ckpts.SkipLoRA	Wraps a partial loader to not restore the LoRA weights.

Function

gm.ckpts.load_params	Restore the params from a checkpoint.
gm.ckpts.save_params	Save the params to a checkpoint.