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ê está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!
Solucionado
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!