from .sg2_loss import StyleGAN2Loss