快捷方式

儲存

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

Storage 是回放緩衝區的容器。

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

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

attach(buffer: Any) None[source]

此函式將取樣器附加到此 storage。

從此 storage 讀取資料的緩衝區必須透過呼叫此方法作為附加實體包含在內。這保證了當 storage 中的資料發生變化時,即使 storage 與其他緩衝區(例如優先順序取樣器)共享,元件也能意識到這些變化。

引數:

buffer – 從此 storage 讀取資料的物件。

dump(*args, **kwargs)[source]

dumps() 的別名。

load(*args, **kwargs)[source]

loads() 的別名。

save(*args, **kwargs)[source]

dumps() 的別名。


© Copyright 2022, Meta.

使用 Sphinx 構建,主題由 Read the Docs 提供。

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

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

檢視教程

資源

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

檢視資源