TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

TensorStorage 的存储检查点器。

此类支持基于 TensorDict 的存储以及 pytrees。

如果提供，此类将调用保存和加载钩子。这些钩子应将要转换的数据以及应保存数据的路径作为输入。