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
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?
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.