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 (Type) – Trainable class for evaluation.

  • config (Dict | None) – Config to pass to Trainable when testing.

  • num_gpus (int) – GPU resources to allocate when testing.

Returns:

True if the save/restore round-trip succeeded.

Return type:

bool

DeveloperAPI: This API may change across minor Ray releases.