Na parte de classificação da aula de funções de perda eu não consigo atribuir o valor de net
para a variável pred
na celula
pred = net(Xtns)
o resultado:
---------------------------------------------------------------------------
NotImplementedError Traceback (most recent call last)
<ipython-input-8-99b085dbb287> in <module>()
----> 1 pred = net(Xtns)
1 frames
/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py in _forward_unimplemented(self, *input)
199 registered hooks while the latter silently ignores them.
200 """
--> 201 raise NotImplementedError
202
203
NotImplementedError:
eu não vi diferença entre meu notebook e o que está no github então não sei o que pode estar dando errado. Eu tentei com CPU, GPU e APU, pensando que poderia ser causado pelo casting mas aparentemente não foi.