TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

TensorStorage 的存储检查点。

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

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