快捷方式

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

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源