ray.train.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 – 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 – Timeout in seconds for this validation task. None is no timeout.

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

Methods

Attributes

fn_kwargs
timeout_s