don't use sigmoid output for tacotron, fix bug for memory queue handling, remove maxout

2026-09-09 11:16:00 +08:00 · 2019-07-22 15:09:05 +02:00
parent ee706b50f6
commit 2bbb3f7a40
3 changed files with 9 additions and 15 deletions
@@ -36,9 +36,7 @@ class Tacotron(nn.Module):
                               forward_attn, trans_agent, forward_attn_mask,
                               location_attn, separate_stopnet)
        self.postnet = PostCBHG(mel_dim)
-        self.last_linear = nn.Sequential(
-            nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim),
-            nn.Sigmoid())
+        self.last_linear = nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim)
        
    def forward(self, characters, text_lengths, mel_specs, speaker_ids=None):
        B = characters.size(0)