dataeval.config

Global configuration settings for DataEval.

Attributes

DeviceLike

Type alias for types that are acceptable for specifying a torch.device.

Functions

get_device([override])

Returns the PyTorch device to use.

get_max_processes()

Returns the maximum number of worker processes to use when running tasks that support parallel processing.

set_device(device)

Sets the default device to use when executing against a PyTorch backend.

set_max_processes(processes)

Sets the maximum number of worker processes to use when running tasks that support parallel processing.

Module Contents

dataeval.config.DeviceLike : TypeAlias = Union[int, str, tuple[str, int], torch.device]

Type alias for types that are acceptable for specifying a torch.device.

See also

torch.device