From 3a99ee9492ec630dbcc1063033ea58e349acc27b Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 20 Nov 2018 14:56:19 +0100 Subject: [PATCH] contiguous on data loader --- datasets/TTSDataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/datasets/TTSDataset.py b/datasets/TTSDataset.py index 4b30710d..a9d111b0 100644 --- a/datasets/TTSDataset.py +++ b/datasets/TTSDataset.py @@ -131,8 +131,8 @@ class MyDataset(Dataset): # convert things to pytorch text_lenghts = torch.LongTensor(text_lenghts) text = torch.LongTensor(text) - linear = torch.FloatTensor(linear) - mel = torch.FloatTensor(mel) + linear = torch.FloatTensor(linear).contiguous() + mel = torch.FloatTensor(mel).contiguous() mel_lengths = torch.LongTensor(mel_lengths) stop_targets = torch.FloatTensor(stop_targets)