ray.train.ValidationTaskConfig

class ray.train.ValidationTaskConfig(fn_kwargs: Dict[str, Any] | None = None, ray_remote_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.

• ray_remote_kwargs – Keyword arguments to pass to ray.remote() for the validation task. This can be used to specify resource requirements, number of retries, etc.

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

Methods

Attributes

fn_kwargs
ray_remote_kwargs