def get_conv_layer(in_channel, out_channel, gan_type='sn_gan', **kwargs):
    if (gan_type == 'sn_gan'):
        return spectral_norm(nn.Conv2d(in_channel, out_channel, **kwargs))
    else:
        return nn.Conv2d(in_channel, out_channel, **kwargs)
def get_conv_layer(in_channel, out_channel, gan_type='sn_gan', **kwargs):
    if (gan_type == 'sn_gan'):
        return spectral_norm(nn.Conv2d(in_channel, out_channel, **kwargs))
    else:
        return nn.Conv2d(in_channel, out_channel, **kwargs)