def device(*, use_cuda, cuda_device):
from lab.util.pytorch import get_device
return get_device(use_cuda, cuda_device)
def device(c: Configs):
from lab.util.pytorch import get_device
return get_device(c.use_cuda, c.cuda_device)