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 notebooks

Used in the tutorials

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.

checkpoint_exists

manager Returns the underlying tf.train.CheckpointManager.

Methods

initialize_or_restore

View source

initialize_or_restore(
    session=None
)

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

save

View source

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

Save state to checkpoint.