StorageEnsemble¶
- class torchrl.data.replay_buffers.StorageEnsemble(*storages: Storage, transforms: list[Transform] = None)[source]¶
一个存储的集合。
此类旨在与
ReplayBufferEnsemble
一起使用。- 参数:
storages (Storage 序列) – 用于组成存储的存储。
- 关键字参数:
transforms (Transform 列表, 可选) – 与 storages 长度相同的转换列表。
警告
此类
get()
的签名与其他存储不匹配,因为它将返回一个元组(buffer_id, samples)
而不仅仅是样本。警告
此类不支持写入(与
WriterEnsemble
类似)。要扩展其中一个回放缓冲区,只需索引父级ReplayBufferEnsemble
对象。- attach(buffer: Any) None ¶
此函数将采样器附加到此存储。
从该存储读取的缓冲区必须通过调用此方法作为附加实体包含在内。这确保了当存储中的数据发生变化时,组件能够感知到变化,即使该存储与其他缓冲区(例如 Priority Samplers)共享。
- 参数:
buffer – 读取此存储的对象。
- dump(*args, **kwargs)¶
dumps()
的别名。
- load(*args, **kwargs)¶
loads()
的别名。
- save(*args, **kwargs)¶
dumps()
的别名。