TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

用于 TensorStorages 的存储检查点器。

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

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