Merge pull request #388 from mittimithai/patch-3

Small fix for "Tacotron" use
This commit is contained in:
Eren Gölge 2020-04-02 13:07:17 +02:00 committed by GitHub
commit 99be88c338
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,