This commit is contained in:
Eren Golge
2018-04-30 06:17:10 -07:00
parent 8319f22278
commit 754e0d3b63
+1 -1
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]