tf_agents.utils.common.Checkpointer

View source on GitHub

Checkpoints training state, policy state, and replay_buffer state.

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

Methods

initialize_or_restore

View source

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

save

View source

Save state to checkpoint.