ray.train.xgboost.XGBoostTrainer.restore#

classmethod XGBoostTrainer.restore(path: str, train_loop_per_worker: Callable[[], None] | Callable[[Dict], None] | None = None, train_loop_config: Dict | None = None, **kwargs) DataParallelTrainer#

从之前中断/失败的运行中恢复一个 DataParallelTrainer。

参数:
  • train_loop_per_worker – 可选的重新指定的训练循环函数。这应该用于重新指定在一个新的 Ray 集群中无法恢复的函数(例如,它持有过时的对象引用)。这应该是传递给原始训练器构造函数的相同训练循环。

  • train_loop_config – 可选的重新指定的训练配置。如果原始的 train_loop_config 包含过时的对象引用,则应同样使用此配置,并且不应从最初传入的内容进行修改。

参见 BaseTrainer.restore() 以获取其他参数的描述。

返回:

调用此方法的 DataParallelTrainer 子类的恢复实例。

返回类型:

DataParallelTrainer