O seguinte erro aconteceu durante execução do cálculo da loss:
loss = criterion(pred, Ytns)
print(loss)RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'
Solucionado
O seguinte erro aconteceu durante execução do cálculo da loss:
loss = criterion(pred, Ytns)
print(loss)RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'
Solução:
No vídeo, o tensor dos targets foi automaticamente transformado como int64, porém na minha máquina/versão foi necessário transformar manualmente para Long.
Xtns = torch.from_numpy(data).float()
Ytns = torch.from_numpy(target).to(torch.long) #modificação