summaryrefslogtreecommitdiff
path: root/become_yukarin/model.py
diff options
context:
space:
mode:
authorHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2017-12-24 20:26:32 +0900
committerHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2017-12-24 20:27:25 +0900
commit3b38bf420774f2a7f718be927689b67446e680c9 (patch)
tree8c18c84042e500e5ff78729a10d21481f7bd4903 /become_yukarin/model.py
parent93df4c160b8332a4ef41190860b5056905143def (diff)
separate-noise-level
Diffstat (limited to 'become_yukarin/model.py')
-rw-r--r--become_yukarin/model.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/become_yukarin/model.py b/become_yukarin/model.py
index 8879f11..8a6af14 100644
--- a/become_yukarin/model.py
+++ b/become_yukarin/model.py
@@ -285,5 +285,8 @@ def create(config: ModelConfig):
aligner = create_aligner(config)
else:
aligner = None
- discriminator = create_discriminator(config.discriminator)
+ if config.discriminator is not None:
+ discriminator = create_discriminator(config.discriminator)
+ else:
+ discriminator = None
return predictor, aligner, discriminator