apply sigmoid to outputs

This commit is contained in:
Eren committed 2018-09-19 15:49:42 +02:00
1 parent e7278437ee
commit f60e4497a6
1 file changed
+1
+1
View File
@@ -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(