From 3293d4e05f3a790d1e3c7011cf459d5bb541ebc7 Mon Sep 17 00:00:00 2001 From: erogol Date: Thu, 2 Apr 2020 13:06:19 +0200 Subject: [PATCH] bug fix to use tacotron --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 0bd1f2af..1fc997ef 100644 --- a/train.py +++ b/train.py @@ -47,7 +47,7 @@ def setup_loader(ap, r, is_val=False, verbose=False): dataset = MyDataset( r, c.text_cleaner, - compute_linear_spec=True if c.model.lower() is 'tacotron' else False, + compute_linear_spec=True if c.model.lower() == 'tacotron' else False, meta_data=meta_data_eval if is_val else meta_data_train, ap=ap, tp=c.characters if 'characters' in c.keys() else None,