def __init__()

in src/image_gen_aux/preprocessors/lineart/model.py [0:0]

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

    def __init__(self, in_features: int):
        super(ResidualBlock, self).__init__()

        conv_block = [
            nn.ReflectionPad2d(1),
            nn.Conv2d(in_features, in_features, 3),
            norm_layer(in_features),
            nn.ReLU(inplace=True),
            nn.ReflectionPad2d(1),
            nn.Conv2d(in_features, in_features, 3),
            norm_layer(in_features),
        ]

        self.conv_block = nn.Sequential(*conv_block)