TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

TensorStorage 的存储检查点。

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

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