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

small bug fixes

Browse Source
This commit is contained in:
Eren Golge
2018-02-09 05:39:58 -08:00
parent 7d5bcd6ca4
commit dadefb5dbc
6 changed files with 51 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
datasets/LJSpeech.py
+4 -3
View File
@@ -72,14 +72,15 @@ class LJSpeechDataset(Dataset):
timesteps = mel.shape[2]
# PAD with zeros that can be divided by outputs per step
# if timesteps % self.outputs_per_step != 0:
linear = pad_per_step(linear, self.outputs_per_step)
mel = pad_per_step(mel, self.outputs_per_step)
if timesteps % self.outputs_per_step != 0:
linear = pad_per_step(linear, self.outputs_per_step)
mel = pad_per_step(mel, self.outputs_per_step)
# reshape jombo
linear = linear.transpose(0, 2, 1)
mel = mel.transpose(0, 2, 1)
# convert things to pytorch
text_lenghts = torch.LongTensor(text_lenghts)
text = torch.LongTensor(text)
linear = torch.FloatTensor(linear)