No artigo "Primeiros Passos com Pytorch" no link abaixo:
https://www.alura.com.br/artigos/primeiros-passos-com-pytorch
Ao executar o trecho de código:
for i in range(epocas):
  y_predito = modelo_classificacao.forward(X_treino)
  custo = funcao_objetivo(y_predito, y_treino)
  custos.append(custo)
  otimizador.zero_grad()
  custo.backward()
  otimizador.step()Retorna o erro:
NotImplementedError                       Traceback (most recent call last)
<ipython-input-29-3704faac9f5e> in <module>()
      1 for i in range(epocas):
----> 2   y_predito = modelo_classificacao.forward(X_treino)
      3   custo = funcao_objetivo(y_predito, y_treino)
      4   custos.append(custo)
      5 
/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: Este erro acontece mesmo rodando o código versionado no github do artigo.
 
            