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