ray.tune.Trainable.restore#

Trainable.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的次要版本之间发生变化。