Solucionado (ver solução)
Solucionado
(ver solução)
2
respostas

Arquivo .ipynb aula 4

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'

2 respostas
solução!

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!