def __init__()

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

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

    def __init__(self, in_features, out_features, stride, use_ac=False):
        super(SingleConvBlock, self).__init__()
        self.use_ac = use_ac
        self.conv = nn.Conv2d(in_features, out_features, 1, stride=stride, bias=True)
        if self.use_ac:
            self.smish = Smish()