No código da quarta aula temos o seguinte código:
if torch.cuda.is_available():
args['device'] = torch.device('cuda')
else:
args['device'] = torch.device('gpu')
Acredito que no else deveria ser 'cpu' e não 'gpu'
No código da quarta aula temos o seguinte código:
if torch.cuda.is_available():
args['device'] = torch.device('cuda')
else:
args['device'] = torch.device('gpu')
Acredito que no else deveria ser 'cpu' e não 'gpu'
Você tem toda razão :) Entrarei em contato para solicitar a correção.
Obrigado Rodrigo, correção feita no GitHub!
Abraços e bons estudos!