ray.tune.utils.validate_save_restore#

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

Helper method to check if your Trainable class will resume correctly.

Parameters:

  • trainable_cls – Trainable class for evaluation.

  • config – Config to pass to Trainable when testing.

  • num_gpus – GPU resources to allocate when testing.

  • use_object_store – Whether to save and restore to Ray’s object store. Recommended to set this to True if planning to use algorithms that pause training (i.e., PBT, HyperBand).

DeveloperAPI: This API may change across minor Ray releases.