def __init__(self, c1, n_classes, upscale_factor, is_aux=False) -> None:
    super().__init__()
    ch = (256 if is_aux else 64)
    c2 = ((n_classes * upscale_factor) * upscale_factor)
    self.conv_3x3 = ConvModule(c1, ch, 3, 1, 1)
    self.conv_1x1 = nn.Conv2d(ch, c2, 1, 1, 0)
    self.upscale = nn.PixelShuffle(upscale_factor)
def __init__(self, c1, n_classes, upscale_factor, is_aux=False) -> None:
    super().__init__()
    ch = (256 if is_aux else 64)
    c2 = ((n_classes * upscale_factor) * upscale_factor)
    self.conv_3x3 = ConvModule(c1, ch, 3, 1, 1)
    self.conv_1x1 = nn.Conv2d(ch, c2, 1, 1, 0)
    self.upscale = nn.PixelShuffle(upscale_factor)