forked from mrq/DL-Art-School
10 lines
284 B
Python
10 lines
284 B
Python
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
|