Reference / API reference / dorothy / dorothy.utils

dorothy.utils

Small shared utilities.

Functions

resolve_device([device])

Resolve a device specification to a torch.device.

dorothy.utils.resolve_device(device='auto')

Resolve a device specification to a torch.device.

Parameters:

device (str | device) – “auto” (CUDA if available, else Apple MPS, else CPU), a device string such as “cpu” / “cuda” / “mps”, or a torch.device.

Returns:

The resolved torch.device.

Return type:

device