快捷方式

NestedStorageCheckpointer

class torchrl.data.replay_buffers.NestedStorageCheckpointer(done_keys=None, reward_keys=None, **kwargs)[source]

以紧凑的形式保存存储,节省 TED 格式的空间并使用内存映射的嵌套张量。

此类显式假设并且不检查

  • 根目录中的 done 状态(包括 terminated 和 truncated)始终为 False;

  • “next”tensordict 中的 observations 向前移动一步(例如,当使用多步转换时则不是这样)。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源