do not limit the possible choices of torch device to use