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