PPOStateCheckpointFile

PPOStateCheckpointFile(state: PPOState, compress: bool = True) -> None

Checkpoint file for saving and restoring the PPO experiment state.

Initialize the PPOStateCheckpointFile.

Parameters:

• state (PPOState) – The PPO experiment state to save.
• compress (bool) – Whether to zstd-compress the checkpoint file.

Source

Properties

state

state: PPOState

Get the experiment state.

Source