快捷方式

NestedStorageCheckpointer

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

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

此类明确假定且不检查以下情况:

  • 根层级的 done 状态(包括 terminated 和 truncated)始终为 False;

  • “next” tensordict 中的观测值在未来被向前偏移一步(例如,在使用多步 transform 时,情况并非如此)。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取供初学者和高级开发者使用的深入教程

查看教程

资源

查找开发者资源并获得问题解答

查看资源