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