From f60e4497a6d586fa4bd49cd4cd6c46ea3346c60e Mon Sep 17 00:00:00 2001
From: Eren <erengolge@gmail.com>
Date: Sat, 15 Sep 2018 18:05:34 +0200
Subject: [PATCH] apply sigmoid to outputs

---
 layers/tacotron.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/layers/tacotron.py b/layers/tacotron.py
index 6ab06d7f..835abde9 100644
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@@ -374,6 +374,7 @@ class Decoder(nn.Module):
             decoder_output = decoder_input
             # predict mel vectors from decoder vectors
             output = self.proj_to_mel(decoder_output)
+            output = torch.sigmoid(output)
             stop_input = output
             # predict stop token
             stop_token, stopnet_rnn_hidden = self.stopnet(