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

bug fix tacotron2, decoder return order fixed

Browse Source
This commit is contained in:
Eren Golge
2019-10-29 13:32:20 +01:00
parent 5a56a2c096
commit 89ef71ead8
3 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/tacotron2.py
+1 -1
View File
@@ -66,7 +66,7 @@ class Tacotron2(nn.Module):
encoder_outputs = self.encoder(embedded_inputs, text_lengths)
encoder_outputs = self._add_speaker_embedding(encoder_outputs,
speaker_ids)
decoder_outputs, stop_tokens, alignments = self.decoder(
decoder_outputs, alignments, stop_tokens = self.decoder(
encoder_outputs, mel_specs, mask)
postnet_outputs = self.postnet(decoder_outputs)
postnet_outputs = decoder_outputs + postnet_outputs