dataeval.config.get_device¶

dataeval.config.get_device(override=None)¶

Returns the PyTorch device to use.

Parameters:¶

override : DeviceLike or None, default None¶

The user specified override if provided, otherwise returns the default device.

Return type:¶

torch.device