Как избежать создания структуры сети должным образом

Question

Я попытался определить функцию feedforwad в моей модели нейронной сети:

class FeedForward(nn.Module):
    def __init__(self):
        super(FeedForward,self).__init__() 
        self.fc1 = nn.Linear(784, 256)
        self.fc2 = nn.Linear(256, 64)
        self.fc2 = nn.Linear(64, 10)

    def feedforward(self, x):
        x = x.view(x.shape[0], -1)  # make sure inputs are flattened 

        x = F.relu(self.fc1(x))
        x = F.relu(self.fc2(x))
        x = F.relu(self.fc3(x))
        x= F.log_softmax(x, dim=1)  # preserve batch dim

        return x

В сообщении говорится:

NotImplementedError

Яне уверен, что мне не хватает.

Answer 1

Имя метода должно быть forward, а не feedforward:

class FeedForward(nn.Module):
    def __init__(self):
        super(FeedForward,self).__init__() 
        self.fc1 = nn.Linear(784, 256)
        self.fc2 = nn.Linear(256, 64)
        self.fc2 = nn.Linear(64, 10)

    def forward(self, x):  # this is what pytorch expects
        x = x.view(x.shape[0], -1)  # make sure inputs are flattened 

        x = F.relu(self.fc1(x))
        x = F.relu(self.fc2(x))
        x = F.relu(self.fc3(x))
        x= F.log_softmax(x, dim=1)  # preserve batch dim

        return x