ray.tune.utils.validate_save_restore#

ray.tune.utils.validate_save_restore(trainable_cls: Type, config: Dict | None = None, num_gpus: int = 0)[源代码]#

帮助方法,用于检查您的可训练类是否能正确恢复。

参数:
  • trainable_cls – 用于评估的可训练类。

  • config – 测试时传递给 Trainable 的配置。

  • num_gpus – 测试时分配的GPU资源。

  • use_object_store – 是否保存并恢复到 Ray 的对象存储。如果计划使用暂停训练的算法(即 PBT、HyperBand),建议将此设置为 True。

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