coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove stop token prediciton

pull/10/head
Eren Golge 2018-03-22 12:50:26 -07:00
parent a925c9c75c
commit 0f3b2ddd7b
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
models/tacotron.py
View File

@ -35,7 +35,7 @@ class Tacotron(nn.Module):
encoder_outputs = self.encoder(inputs)
# (B, T', mel_dim*r)
mel_outputs, alignments, stop_outputs = self.decoder(
mel_outputs, alignments = self.decoder(
encoder_outputs, mel_specs)
# Post net processing below
@ -43,9 +43,8 @@ class Tacotron(nn.Module):
# Reshape
# (B, T, 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.last_linear(linear_outputs)
return mel_outputs, linear_outputs, alignments, stop_outputs
return mel_outputs, linear_outputs, alignments