Data loader bug fix 2

2026-09-10 11:50:20 +08:00 · 2018-03-26 11:07:15 -07:00
parent 3c084177c6
commit 1ff8d6d2b7
4 changed files with 9 additions and 5 deletions
@@ -125,6 +125,7 @@ class LJSpeechDataset(Dataset):
            mel = torch.FloatTensor(mel)
            mel_lengths = torch.LongTensor(mel_lengths)
            stop_targets = torch.FloatTensor(stop_targets)
+            
            return text, text_lenghts, linear, mel, mel_lengths, stop_targets, item_idxs[0]

        raise TypeError(("batch must contain tensors, numbers, dicts or lists;\