快速鍵

TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer[原始碼]

TensorStorages 的儲存檢查點。

此類別支援基於 TensorDict 的儲存以及 pytrees。

如果提供,此類別將呼叫儲存和載入 Hook。 這些 Hook 應將正在轉換的資料以及應儲存資料的路徑作為輸入。

文件

存取 PyTorch 的全面開發者文件

檢視文件

教學

取得適合初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得解答

檢視資源