def __init__(self, image_size, in_channels, *args, **kwargs):
    super().__init__(image_size, (in_channels * 2), *args, **kwargs)
def __init__(self, image_size, in_channels, *args, **kwargs):
    super().__init__(image_size, (in_channels * 2), *args, **kwargs)