def __init__()

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


    def __init__(self, in_ch, out_ch):
        super(DoubleFusion, self).__init__()
        self.DWconv1 = nn.Conv2d(in_ch, in_ch * 8, kernel_size=3, stride=1, padding=1, groups=in_ch)  # before 64
        self.PSconv1 = nn.PixelShuffle(1)

        self.DWconv2 = nn.Conv2d(24, 24 * 1, kernel_size=3, stride=1, padding=1, groups=24)  # before 64  instead of 32

        self.AF = Smish()  # XAF() #nn.Tanh()# XAF() #   # Smish()#