From 0f3b2ddd7b2ef4e5fdae4c535fa6288de704beff Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 22 Mar 2018 12:50:26 -0700 Subject: [PATCH] remove stop token prediciton --- models/tacotron.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/models/tacotron.py b/models/tacotron.py index a8b04fbd..a485d68e 100644 --- a/models/tacotron.py +++ b/models/tacotron.py @@ -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