gm.ckpts.save_params

gm.ckpts.save_params#

gemma.gm.ckpts.save_params(
params: collections.abc.Mapping[str, Any],
path: str | os.PathLike,
*,
wait_until_finished: bool = False,
) → None[source]

Save the params to a checkpoint.

Parameters:

  • params – The params to save.

  • path – The directory to which save the checkpoint.

  • wait_until_finished – If True, waits for the checkpoint save to complete before returning.