From aff1b5313be74dd7f34ad2475c1e43568d92886f Mon Sep 17 00:00:00 2001 From: babysor00 Date: Wed, 17 Nov 2021 00:12:27 +0800 Subject: [PATCH] Order of declared pytorch module matters --- synthesizer/models/tacotron.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/synthesizer/models/tacotron.py b/synthesizer/models/tacotron.py index eaa7818..431fcb2 100644 --- a/synthesizer/models/tacotron.py +++ b/synthesizer/models/tacotron.py @@ -343,8 +343,9 @@ class Tacotron(nn.Module): 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,