reference/generated/torchrl.trainers.algorithms.configs.checkpoint.CheckpointRotationConfig

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

torchrl.trainers.algorithms.configs.checkpoint.CheckpointRotationConfig

class torchrl.trainers.algorithms.configs.checkpoint.CheckpointRotationConfig(directory: str, keep_last: int, keep_best: list[str] | None = None, prefix: str = 'checkpoint', _target_: str = 'torchrl.checkpoint.CheckpointRotation')

Hydra configuration for CheckpointRotation.

keep_best is a two-item list [metadata_key, mode] with mode one of "min" or "max".

See also

CheckpointRotation