H5StorageCheckpointer

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

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

此類明確假定並 不檢查 以下情況：

• 根級別的 done 狀態（包括 terminated 和 truncated）總是為 False；

• “next” tensordict 中的觀察值在時間上向前偏移一步（例如，使用多步 transforms 時則不是這種情況）。

關鍵字引數：

• checkpoint_file – 用於儲存檢查點資料的檔名。如果傳遞給 dumps / loads 的路徑以 .h5 字尾結尾，則此引數將被忽略。預設為 "checkpoint.h5"。

• h5_kwargs (Dict[str, Any] 或 Tuple[Tuple[str, Any], ...]) – 要傳遞給 h5py.File.create_dataset() 的關鍵字引數。

注意

為防止記憶體不足問題，H5 檔案的資料將暫時寫入儲存在共享檔案系統中的記憶體對映張量上。因此，載入過程中物理記憶體使用量可能會增加。