Shortcuts

TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer[source]

A storage checkpointer for TensorStorages.

This class supports TensorDict-based storages as well as pytrees.

This class will call save and load hooks if provided. These hooks should take as input the data being transformed as well as the path where the data should be saved.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources