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