from models.styled_sr.discriminator import StyleSrGanDivergenceLoss
def create_stylesr_loss(opt_loss, env):
type = opt_loss['type']
if type == 'style_sr_gan_divergence_loss':
return StyleSrGanDivergenceLoss(opt_loss, env)
else:
raise NotImplementedError
