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 文件的数据将暂时写入存储在共享文件系统中的内存映射张量上。因此，加载过程中物理内存使用量可能会增加。