tf_agents.utils.common.Checkpointer

Checkpoints training state, policy state, and replay_buffer state.

tf_agents.utils.common.Checkpointer(
    ckpt_dir, max_to_keep=20, **kwargs
)

Used in the tutorials
Checkpointer and PolicySaver

Args
`ckpt_dir`	The directory to save checkpoints.
`max_to_keep`	Maximum number of checkpoints to keep (if greater than the max are saved, the oldest checkpoints are deleted).
`**kwargs`	Items to include in the checkpoint.

Attributes
`checkpoint_exists`
`manager`	Returns the underlying tf.train.CheckpointManager.

Attributes

checkpoint_exists

manager Returns the underlying tf.train.CheckpointManager.

initialize_or_restore(
    session=None
)

Initialize or restore graph (based on checkpoint if exists).

save(
    global_step: tf.Tensor, options: tf.train.CheckpointOptions = None
)

Save state to checkpoint.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.