From bf7590c7d254cd5569332a3554fe643d4f648a2c Mon Sep 17 00:00:00 2001 From: Eren Date: Thu, 4 Oct 2018 13:53:39 +0200 Subject: [PATCH] Bug fix --- datasets/LJSpeech.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/datasets/LJSpeech.py b/datasets/LJSpeech.py index e72a66e0..495283e1 100644 --- a/datasets/LJSpeech.py +++ b/datasets/LJSpeech.py @@ -139,8 +139,7 @@ class MyDataset(Dataset): 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] + return text, text_lenghts, linear, mel, mel_lengths, stop_targets, item_idxs raise TypeError(("batch must contain tensors, numbers, dicts or lists;\ found {}".format(type(batch[0]))))