Small fix for "Tacotron" use

"Tacotron" won't work without this fix, since the linear spectrograms end up not getting computed
This commit is contained in:
mittimithai
2020-04-01 11:57:53 -07:00
committed by GitHub
parent 141797b6ae
commit 6501369b0a
+1 -1
View File
@@ -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,