H5StorageCheckpointer

class torchrl.data.replay_buffers.H5StorageCheckpointer(*, checkpoint_file: str = 'checkpoint.h5', done_keys=None, reward_keys=None, h5_kwargs=None, **kwargs)

以緊湊的形式儲存儲存體，節省 TED 格式的空間並使用 H5 格式儲存資料。

此類別明確假設且 *不* 檢查

• 根目錄下的完成狀態（包括終止和截斷）始終為 False；

• “next” tensordict 中的觀察結果在未來移動一個步驟（例如，使用多步驟轉換時並非如此）。

關鍵字引數:

• checkpoint_file – 要儲存檢查點資料的檔案名稱。 如果傳遞給 dumps / loads 的路徑以 .h5 後綴結尾，則會忽略此參數。 預設值為 "checkpoint.h5"。

• h5_kwargs (Dict[str, Any] 或 Tuple[Tuple[str, Any], ...]) – 要傳遞給 h5py.File.create_dataset() 的 kwargs。

注意

為了防止記憶體不足問題，H5 檔案的資料將暫時寫入儲存在共用檔案系統中的記憶體對應張量。 因此，在載入期間，實體記憶體使用量可能會增加。