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()#