def forward()

in train-cnn/Container/scripts/multi-train.py [0:0]

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

    def forward(self, x: torch.Tensor):
        x = self.conv_layer_1(x)
        x = self.conv_layer_2(x)
        for _ in range(4):
            x = self.conv_layer_3(x)
        x = self.classifier(x)
        return x