apply sigmoid to outputs

2026-08-29 11:14:04 +08:00 · 2018-09-19 15:49:42 +02:00
parent e7278437ee
commit f60e4497a6
1 changed files with 1 additions and 0 deletions
@@ -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(