e.g. device may need to be set to 'cpu'
e.g. device may need to be set to 'cpu'