快捷方式

NestedStorageCheckpointer

class torchrl.data.replay_buffers.NestedStorageCheckpointer(done_keys=None, reward_keys=None, **kwargs)[原始檔]

以緊湊形式儲存儲存內容,節省 TED 格式的空間,並使用記憶體對映的巢狀張量。

此類明確假定且不檢查以下情況:

  • 根層級的 done 狀態(包括 terminated 和 truncated)始終為 False;

  • “next” tensordict 中的觀測值在未來被向前偏移一步(例如,在使用多步 transform 時,情況並非如此)。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取供初學者和高階開發者使用的深入教程

檢視教程

資源

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

檢視資源