def forward()

in fast_grad_models.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def forward(self, x):
        for _, conv in enumerate(self.conv_layers):
            x = F.max_pool2d(F.relu(conv(x)), self.pool_size, self.pool_size)
        out = x.view(x.size(0), -1)
        if self.normalize:
            out = F.normalize(out)
        return out