快捷方式

Storage

class torchrl.data.replay_buffers.Storage(max_size: int, checkpointer: StorageCheckpointerBase | None = None, compilable: bool = False)[原始碼]

Storage 是經驗回放緩衝區 (replay buffer) 的容器。

每個 storage 都必須實現 set、get 和 __len__ 方法。Get 和 set 方法應該支援整數以及整數列表。

storage 不需要有固定的大小,但如果它有固定大小,則應確保其與緩衝區大小相容。

attach(buffer: Any) None[原始碼]

此函式將取樣器附加到此儲存。

從該儲存讀取的緩衝區必須透過呼叫此方法作為已附加實體包含進來。這確保了當儲存中的資料發生變化時,元件能夠感知到這些變化,即使該儲存與其他緩衝區(例如,Priority Samplers)共享。

引數:

buffer – 讀取此儲存的物件。

dump(*args, **kwargs)[原始碼]

dumps() 的別名。

load(*args, **kwargs)[原始碼]

loads() 的別名。

save(*args, **kwargs)[原始碼]

dumps() 的別名。

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

為初學者和高階開發者提供深入的教程

檢視教程

資源

查詢開發資源並讓您的問題得到解答

檢視資源