From 3b38bf420774f2a7f718be927689b67446e680c9 Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Sun, 24 Dec 2017 20:26:32 +0900 Subject: separate-noise-level --- become_yukarin/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'become_yukarin/model.py') 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 -- cgit v1.2.3-70-g09d2