From 2563fb873eaaa54f6539012d1c7e9fad12fb474b Mon Sep 17 00:00:00 2001 From: Eugene Ingerman Date: Sat, 24 Aug 2019 14:17:20 -0700 Subject: [PATCH] Fixed postnet for GST. --- models/tacotrongst.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/models/tacotrongst.py b/models/tacotrongst.py index 5b372338..c18a5e98 100644 --- a/models/tacotrongst.py +++ b/models/tacotrongst.py @@ -38,9 +38,8 @@ class TacotronGST(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)