reference/generated/torchrl.checkpoint.CheckpointAdapter

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

CheckpointAdapter

class torchrl.checkpoint.CheckpointAdapter

Interface used to save and restore one checkpoint component.

Adapters receive a real local directory even when the outer checkpoint is an archive. Custom adapters can therefore write any number of files without depending on the container implementation.

Examples

>>> from torchrl.checkpoint import CheckpointAdapter
>>> issubclass(CheckpointAdapter, object)
True

abstract load(component: Any, path: Path, *, map_location: Any, tensor_load_kwargs: Mapping[str, Any], args: tuple[Any, ...], kwargs: Mapping[str, Any]) → Any

Restore component from path and optionally return a value.

abstract save(component: Any, path: Path, *, args: tuple[Any, ...], kwargs: Mapping[str, Any]) → Any

Save component below path.