From 6501369b0a2c32e896900d913f31f10b583045b4 Mon Sep 17 00:00:00 2001
From: mittimithai <mittimithai@gmail.com>
Date: Wed, 1 Apr 2020 11:57:53 -0700
Subject: [PATCH] Small fix for "Tacotron" use

"Tacotron" won't work without this fix, since the linear spectrograms end up not getting computed
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 62800c7b..a4aaed17 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 in ["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,