Order of declared pytorch module matters

c-lins · Nov 16, 2021 · aff1b53 · aff1b53
1 parent 7dca74e
commit aff1b53
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/synthesizer/models/tacotron.py b/synthesizer/models/tacotron.py
@@ -343,8 +343,9 @@ def __init__(self, embed_dims, num_chars, encoder_dims, decoder_dims, n_mels,
         project_dims = encoder_dims + speaker_embedding_size
         if hparams.use_gst: 
             project_dims += gst_hp.E
-            self.gst = GlobalStyleToken(speaker_embedding_size)
         self.encoder_proj = nn.Linear(project_dims, decoder_dims, bias=False)
+        if hparams.use_gst: 
+            self.gst = GlobalStyleToken(speaker_embedding_size)
         self.decoder = Decoder(n_mels, encoder_dims, decoder_dims, lstm_dims,
                                dropout, speaker_embedding_size)
         self.postnet = CBHG(postnet_K, n_mels, postnet_dims,