ray.tune.utils.validate_save_restore#

ray.tune.utils.validate_save_restore(trainable_cls: Type, config: Dict | None = None, num_gpus: int = 0) → bool[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.

Returns:

True if the save/restore round-trip succeeded.

DeveloperAPI: This API may change across minor Ray releases.