ValidationTaskConfig

class ray.train.ValidationTaskConfig(fn_kwargs: Dict[str, Any] | None = None, timeout_s: float | None = None)

Bases: object

Configuration for a specific validation task, passed to report().

Parameters:

• fn_kwargs (Dict[str, Any] | None) – json-serializable keyword arguments to pass to the validation function. Note that we always pass checkpoint as the first argument to the validation function.

• timeout_s (float | None) – Timeout in seconds for this validation task. In ValidationConfig.task_config, None means no timeout. In a ValidationTaskConfig passed to report(), None uses ValidationConfig.task_config.timeout_s.

PublicAPI (alpha): This API is in alpha and may change before becoming stable.

Methods

Attributes

fn_kwargs
timeout_s