dataeval.config.set_device¶

dataeval.config.set_device(device)¶

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

Parameters:¶

device : DeviceLike or None¶

The default device to use. See documentation for more information.

Return type:¶

None

See also

torch.device