Solucionado (ver solução)
Solucionado
(ver solução)
1
resposta

Durante a classificação, meu predict está errado

Eu fiz o passo a passo do vídeo junto com a professora, e inclusive conferi o código lá no github, mas tem alguma coisa errada com o meu predict da rede durante a parte de classificação. Quando eu rodo essa parte:

pred = net(Xtns)
print(pred.shape)

aparece o seguinte erro: "AttributeError: 'int' object has no attribute 'shape'", o que não faz sentido. Daí quando eu dou print no pred, a resposta é 3. Não sei o que estou fazendo errado. Será que alguém pode me ajudar?

1 resposta
solução!

Encontrei meu erro, na função forward eu estava retornando o outpu_size ao invés do output. Errinho besta que eu não estava conseguindo ver antes.