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?