ray.rllib.algorithms.algorithm.Algorithm.restore#

Algorithm.restore(checkpoint_path: str | Checkpoint | _TrainingResult)#

从给定的模型检查点恢复训练状态。

这些检查点是从调用 save() 返回的。

子类应重写 load_checkpoint() 以恢复状态。此方法恢复与检查点一起保存的额外元数据。

checkpoint_path 应与 save() 的返回值匹配。

参数:

  • checkpoint_path – 恢复检查点的路径。如果此路径在本地节点上不存在，如果可用，它将从外部（云）存储中获取，或者从远程节点恢复。

  • checkpoint_node_ip – 如果提供，尝试从该节点恢复检查点，如果它在本地或云存储中不存在。

  • fallback_to_latest – 如果为 True，如果找不到给定的 checkpoint_path，将尝试恢复最新的可用检查点。

开发者API: 此API可能会在Ray的次要版本之间发生变化。