gm.ckpts.load_params

gemma.gm.ckpts.load_params(

path: epath.PathLike,

*,

params: params_lib.Params | None = None,

donate: bool = True,

text_only: bool = False,

sharding: kd.sharding.ShardingTree | None = None,

quantize: bool = False,

) → params_lib.Params

从检查点恢复参数 (params)。

参数 (Parameters)：

• path – orbax 检查点的路径。

• params – 与检查点结构匹配的状态。用于使用正确的切分 (sharding) 恢复参数 (params)。

• donate – 如果为 True 且提供了 params，则将释放来自参数 (params) 的内存。

• text_only – 如果为 True，则仅恢复文本参数 (params)，并忽略多模态参数 (params)。

• sharding – 如果提供，参数 (params) 将使用此切分 (sharding) 恢复。这与 params 互斥。

• quantize – 如果为 True，参数 (params) 将被映射以启用量化感知训练 (quantization aware training)。

返回 (Returns)：

恢复的参数 (params)。