gm.ckpts

[[Source]]

检查点 API。

符号

类

gm.ckpts.AnchoredPolicyLoader	用于 gm.nn.AnchoredPolicy 模型的加载器。
gm.ckpts.CheckpointPath	Gemma 检查点的硬编码路径。
gm.ckpts.LoadCheckpoint	从 Gemma 检查点加载权重。
gm.ckpts.SkipLoRA	包装一个部分加载器以不恢复 LoRA 权重。

函数

gm.ckpts.load_params	从检查点恢复参数。
gm.ckpts.save_params	将参数保存到检查点。