From 754e0d3b63a64a06815ff425c07df391aef774c5 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 30 Apr 2018 06:17:10 -0700 Subject: [PATCH] bug fix --- datasets/LJSpeech.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/LJSpeech.py b/datasets/LJSpeech.py index aea6575d..e3f78176 100644 --- a/datasets/LJSpeech.py +++ b/datasets/LJSpeech.py @@ -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]