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