DL-Art-School/codes/models/stylegan/__init__.py

11 lines
377 B
Python
Raw Normal View History

2020-12-18 16:24:31 +00:00
import models.stylegan.stylegan2_lucidrains as stylegan2
def create_stylegan2_loss(opt_loss, env):
type = opt_loss['type']
if type == 'stylegan2_divergence':
2020-11-15 18:32:35 +00:00
return stylegan2.StyleGan2DivergenceLoss(opt_loss, env)
elif type == 'stylegan2_pathlen':
2020-11-15 18:32:35 +00:00
return stylegan2.StyleGan2PathLengthLoss(opt_loss, env)
else:
raise NotImplementedError