ray.train.ValidationTaskConfig

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

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.

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

Methods

Attributes

fn_kwargs