Erro NotImplementedError

Oi, Abel! Tudo bem com você?

Esse erro pode surgir devido a um problema na construção da classe Modelo. Você pode confirmar para mim se o seu código da classe Modelo tem a seguinte estrutura:

class Modelo(nn.Module):
    def __init__ (self,entrada=7,camada_escondida1=14,camada_escondida2=49,saida=3):
        super().__init__()
        self.fc1 = nn.Linear(entrada,camada_escondida1)
        self.fc2 = nn.Linear(camada_escondida1, camada_escondida2)
        self.out = nn.Linear(camada_escondida2, saida)
    def forward(self, x):
        x = F.relu(self.fc1(x))
        x = F.relu(self.fc2(x))
        x = self.out(x)
        return x

Pela forma separada de apresentar as funções da classe Modelo, pode ocorrer uma confusão na implementação do código, deixando alguma das funções em um local incorreto. Se algumas das funções do seu código não manterem a estrutura colocada acima, você pode, por gentileza, substituir no seu código e verificar se assim funciona?

Fico no aguardo do seu retorno!