compute and add style tokens in gst

2026-09-11 12:00:24 +08:00 · 2019-09-21 09:58:58 +02:00
parent 6561013d28
commit 1f4ec804b6
2 changed files with 2 additions and 1 deletions
@@ -40,7 +40,7 @@ class TacotronGST(Tacotron):
                        location_attn,
                        separate_stopnet)
        gst_embedding_dim = 256
-        decoder_dim = 512 + gst_embedding_dim if num_speakers > 1 else 256 + gst_embedding_dim
+        decoder_dim = 512 if num_speakers > 1 else 256 
        proj_speaker_dim = 80 if num_speakers > 1 else 0
        self.decoder = Decoder(decoder_dim, mel_dim, r, memory_size, attn_win,
                               attn_norm, prenet_type, prenet_dropout,