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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

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,