parsnip.parse_device

parsnip.parse_device(device)

Figure out which PyTorch device to use

Parameters:

device (str) – Requested device

Returns:

Device to use

Return type:

str