Consistent usage of 'cuda' or 'gpu' #136
Description
Could you change the device specifications to one consistent name, either cuda cuda or gpu, although I would suggest using cuda since this works with pytorch as well.
{{ message }}
Could you change the device specifications to one consistent name, either cuda cuda or gpu, although I would suggest using cuda since this works with pytorch as well.