apply sigmoid to outputs

2018-09-15 18:05:34 +02:00 · 2018-09-15 18:05:34 +02:00 · f60e4497a6
parent e7278437ee
commit f60e4497a6
1 changed files with 1 additions and 0 deletions
--- 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(