def __init__()

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

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

    def __init__(self, in_ch, out_ch):
        super(CoFusion2, self).__init__()
        self.conv1 = nn.Conv2d(in_ch, 32, kernel_size=3, stride=1, padding=1)  # before 64
        self.conv3 = nn.Conv2d(32, out_ch, kernel_size=3, stride=1, padding=1)  # before 64  instead of 32
        self.smish = Smish()  # nn.ReLU(inplace=True)