def forward()

in models/alldnet.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

    def forward(self, x):
        activations = []
        out = F.relu(self.conv1(x))
        out = F.max_pool2d(out, 2)
        # out = out.view(out.size(0), -1)
        # activations.append(out)
        out = F.relu(self.conv2(out))
        # out = out.view(out.size(0), 16, 10, -1)
        out = F.max_pool2d(out, 2)
        out = out.view(out.size(0), -1)
        activations.append(out)
        out = F.relu(self.fc1(out))
        activations.append(out)
        out = F.relu(self.fc2(out))
        activations.append(out)
        out = self.fc3(out)
        return out, activations