reference/generated/torchrl.data.replay_buffers.StorageEnsembleCheckpointer

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

StorageEnsembleCheckpointer

class torchrl.data.replay_buffers.StorageEnsembleCheckpointer

Checkpointer for ensemble storages.

register_load_hook(hook)

Registers a load hook for this checkpointer.

register_save_hook(hook)

Registers a save hook for this checkpointer.