H5StorageCheckpointer

class 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 中的 observation 會提前一步（例如，當使用多步變換時則不是這種情況）。

關鍵字引數:

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

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

注意

為避免記憶體不足的問題，H5 檔案的資料將臨時寫入共享檔案系統中的記憶體對映張量。因此，載入時物理記憶體使用量可能會增加。