ListStorage¶
- class torchrl.data.replay_buffers.ListStorage(max_size: int | None = None, compilable: bool = False)[source]¶
基于列表的存储。
此类不能通过 PyTrees 扩展,在调用
extend()
时提供的数据应为可迭代对象(如列表、元组、张量或具有非空批量大小的 tensordicts)。- 参数:
max_size (int, optional) – 存储中存储的最大元素数量。如果未提供,则创建一个无限大小的存储。
- attach(buffer: Any) None ¶
此函数将采样器附加到此存储。
读取此存储的缓冲区必须通过调用此方法作为附加实体包含在内。这确保了当存储中的数据发生更改时,即使存储与其他缓冲区(例如优先采样器)共享,组件也能感知到更改。
- 参数:
buffer – 从此存储读取数据的对象。
- dump(*args, **kwargs)¶
dumps()
的别名。
- load(*args, **kwargs)¶
loads()
的别名。
- save(*args, **kwargs)¶
dumps()
的别名。