From e7278437ee24b0962f59bb5c8512c29e56eb71f2 Mon Sep 17 00:00:00 2001 From: Eren Date: Thu, 6 Sep 2018 15:27:15 +0200 Subject: [PATCH] Please enter the commit message for your changes. Lines starting --- models/tacotron.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/models/tacotron.py b/models/tacotron.py index cd023a92..8a215b90 100644 --- a/models/tacotron.py +++ b/models/tacotron.py @@ -23,7 +23,9 @@ class Tacotron(nn.Module): self.encoder = Encoder(embedding_dim) self.decoder = Decoder(256, mel_dim, r) self.postnet = PostCBHG(mel_dim) - self.last_linear = nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim) + self.last_linear = nn.Sequential( + nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim), + nn.Sigmoid()) def forward(self, characters, mel_specs=None, mask=None): B = characters.size(0)