快捷方式

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() 的别名。

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

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

查看资源