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

remove stop token prediciton

Browse Source
This commit is contained in:
Eren Golge
2018-03-22 12:50:26 -07:00
parent a925c9c75c
commit 0f3b2ddd7b
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/tacotron.py
+2 -3
View File
@@ -35,7 +35,7 @@ class Tacotron(nn.Module):
encoder_outputs = self.encoder(inputs) encoder_outputs = self.encoder(inputs)
# (B, T', mel_dim*r) # (B, T', mel_dim*r)
mel_outputs, alignments, stop_outputs = self.decoder( mel_outputs, alignments = self.decoder(
encoder_outputs, mel_specs) encoder_outputs, mel_specs)
# Post net processing below # Post net processing below
@@ -43,9 +43,8 @@ class Tacotron(nn.Module):
# Reshape # Reshape
# (B, T, mel_dim) # (B, T, mel_dim)
mel_outputs = mel_outputs.view(B, -1, self.mel_dim) mel_outputs = mel_outputs.view(B, -1, self.mel_dim)
stop_outputs = stop_outputs.view(B, -1)
linear_outputs = self.postnet(mel_outputs) linear_outputs = self.postnet(mel_outputs)
linear_outputs = self.last_linear(linear_outputs) linear_outputs = self.last_linear(linear_outputs)
return mel_outputs, linear_outputs, alignments, stop_outputs return mel_outputs, linear_outputs, alignments