ray.rllib.utils.checkpoints.获取检查点信息

ray.rllib.utils.checkpoints.get_checkpoint_info(checkpoint: str | Checkpoint) → Dict[str, Any]

返回一个包含算法/策略检查点信息的字典。

如果给定的检查点是一个 >=v1.0 的检查点目录，尝试从包含的 rllib_checkpoint.json 文件中读取所有信息。

参数:

checkpoint – 检查点目录（str）或一个 AIR 检查点对象。

返回:

“type”: 可以是 “Policy” 或 “Algorithm”。”checkpoint_version”: 一个版本元组，例如 v1.0，表示检查点版本。这将帮助 RLlib 保持与未来 Ray 和检查点版本的向后兼容性。”checkpoint_dir”: 包含所有检查点文件的目录。这可能与传入的 checkpoint 参数相同。”state_file”: 包含 Algorithm/Policy 状态信息的主文件。这通常是一个 pickle 编码的文件。”policy_ids”: 如果处理的是 Algorithm 检查点，则为可选的 PolicyID 集合。如果 checkpoint 是 Policy 检查点，则为 None。

返回类型:

A dict containing the keys

PublicAPI (alpha): 此API处于alpha阶段，可能在稳定之前发生变化。