快捷方式

ListStorage

class torchrl.data.replay_buffers.ListStorage(max_size: Optional[int] = None)[源代码]

存储在列表中的存储。

此类不能使用 PyTrees 扩展,在调用 extend() 期间提供的数据应该是可迭代对象(如列表、元组、张量或具有非空批大小的 tensordict)。

参数:

max_size (int, 可选) – 存储中存储的最大元素数量。如果未提供,则会创建一个无限制的存储。

attach(buffer: Any) None

此函数将采样器附加到此存储。

从该存储读取数据的缓冲区必须通过调用此方法作为附加实体包含在内。这保证了当存储中的数据发生更改时,即使存储与其他缓冲区(例如,优先级采样器)共享,组件也会意识到更改。

参数:

buffer – 从该存储读取数据的对象。

dump(*args, **kwargs)

别名,指向 dumps()

load(*args, **kwargs)

别名,指向 loads()

save(*args, **kwargs)

别名,指向 dumps()

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

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

查看资源