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

type dataeval.config.DeviceLike = int | str | tuple[str, int] | torch.device

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

See also

torch.device