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