dataeval.config.set_device

dataeval.config.set_device(device)

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

Parameters:
device

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

Return type:

None