wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-10 11:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

contiguous on data loader

Browse Source
This commit is contained in:
Eren Golge
2018-11-20 14:56:19 +01:00
parent 660f8c7b66
commit 3a99ee9492
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
datasets/TTSDataset.py
+2 -2
View File
@@ -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)