TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

TensorStorages 的存储检查点。

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

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