reference/generated/torchrl.data.replay_buffers.StorageCheckpointerBase

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

StorageCheckpointerBase

class torchrl.data.replay_buffers.StorageCheckpointerBase

Public base class for storage checkpointers.

Each storage checkpointer must implement a save and load method that take as input a storage and a path.

register_load_hook(hook)

Registers a load hook for this checkpointer.

register_save_hook(hook)

Registers a save hook for this checkpointer.