mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
8319f22278
commit
754e0d3b63
|
@ -128,7 +128,7 @@ class LJSpeechDataset(Dataset):
|
||||||
linear = torch.FloatTensor(linear)
|
linear = torch.FloatTensor(linear)
|
||||||
mel = torch.FloatTensor(mel)
|
mel = torch.FloatTensor(mel)
|
||||||
mel_lengths = torch.LongTensor(mel_lengths)
|
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]
|
return text, text_lenghts, linear, mel, mel_lengths, stop_targets, item_idxs[0]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue