DumpLoadCheckpointAdapter
-
class torchrl.checkpoint.DumpLoadCheckpointAdapter[source]
Adapter for objects exposing dump(path) and load(path).
Examples
>>> from torchrl.checkpoint import DumpLoadCheckpointAdapter
>>> DumpLoadCheckpointAdapter().adapter_id
'torchrl.dump_load'
-
load(component: Any, path: Path, *, map_location: Any, tensor_load_kwargs: Mapping[str, Any], args: tuple[Any, ...], kwargs: Mapping[str, Any]) → Any[source]
Restore component from path and optionally return a value.
-
save(component: Any, path: Path, *, args: tuple[Any, ...], kwargs: Mapping[str, Any]) → Any[source]
Save component below path.
