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'
Solucionado (ver solução)
Solucionado
(ver solução)
2
respostas
respostas
Arquivo .ipynb aula 4
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!