Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fix

This commit is contained in:
Eren Golge 2018-04-30 06:17:10 -07:00
parent 8319f22278
commit 754e0d3b63
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
datasets/LJSpeech.py
View File

@ -128,7 +128,7 @@ class LJSpeechDataset(Dataset):
linear = torch.FloatTensor(linear)
mel = torch.FloatTensor(mel)
mel_lengths = torch.LongTensor(mel_lengths)
stop_targets = torch.FloatTensor(stop_targets)
stop_targets = torch.FloatTensor(stop_targets).squeeze()
return text, text_lenghts, linear, mel, mel_lengths, stop_targets, item_idxs[0]