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

bug fix for tacotron and tests update

Browse Source
This commit is contained in:
Eren Golge
2019-03-06 13:43:29 +01:00
parent b031a65677
commit a9ce1d4f19
4 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/tacotron.py
+1 -1
View File
@@ -28,7 +28,7 @@ class Tacotron(nn.Module):
nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim),
nn.Sigmoid())
def forward(self, characters, text_lengths, mel_specs=None):
def forward(self, characters, text_lengths, mel_specs):
B = characters.size(0)
mask = sequence_mask(text_lengths).to(characters.device)
inputs = self.embedding(characters)
models/tacotron2.py
+1 -1
View File
@@ -28,7 +28,7 @@ class Tacotron2(nn.Module):
def forward(self, text, text_lengths, mel_specs=None):
# compute mask for padding
mask = sequence_mask(text_lengths).to(characters.device)
mask = sequence_mask(text_lengths).to(text.device)
embedded_inputs = self.embedding(text).transpose(1, 2)
encoder_outputs = self.encoder(embedded_inputs, text_lengths)
mel_outputs, stop_tokens, alignments = self.decoder(