mirror of https://github.com/coqui-ai/TTS.git
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:
parent
141797b6ae
commit
6501369b0a
2
train.py
2
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,
|
||||
|
|
Loading…
Reference in New Issue