def __init__(self, base_width, kernel_size=3):
super(L3SX, self).__init__()
self.in_channels = 3 * base_width
self.out_channels = 4 * base_width
self.upsample = Upsample()
self.conv = conv.ConvELU2D(self.in_channels, self.out_channels,
kernel_size)
def __init__(self, base_width, kernel_size=3):
super(L3SY, self).__init__()
self.in_channels = (4 + 3) * base_width
self.out_channels = 4 * base_width
self.cat = shape.Concatenate()
self.upsample = Upsample()
self.conv = conv.ConvELU2D(self.in_channels, self.out_channels,
kernel_size)