def __init__()

in src/image_gen_aux/preprocessors/teed/teed.py [0:0]

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

    def __init__(self, input_features, out_features):
        super(_DenseLayer, self).__init__()

        (
            self.add_module(
                "conv1",
                nn.Conv2d(
                    input_features,
                    out_features,
                    kernel_size=3,
                    stride=1,
                    padding=2,
                    bias=True,
                ),
            ),
        )
        (self.add_module("smish1", Smish()),)
        self.add_module(
            "conv2",
            nn.Conv2d(out_features, out_features, kernel_size=3, stride=1, bias=True),
        )