StarMirror/MockingBird
1
0
Fork 0
mirror of https://github.com/babysor/MockingBird.git synced 2024-03-22 13:11:31 +08:00
Code Issues Projects Releases Wiki Activity

Order of declared pytorch module matters

Browse Source
This commit is contained in:
babysor00 2021-11-17 00:12:27 +08:00
parent 7dca74e032
commit aff1b5313b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
synthesizer/models/tacotron.py
View File

@ -343,8 +343,9 @@ class Tacotron(nn.Module):
project_dims = encoder_dims + speaker_embedding_size project_dims = encoder_dims + speaker_embedding_size
if hparams.use_gst: if hparams.use_gst:
project_dims += gst_hp.E project_dims += gst_hp.E
self.gst = GlobalStyleToken(speaker_embedding_size)
self.encoder_proj = nn.Linear(project_dims, decoder_dims, bias=False) 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, self.decoder = Decoder(n_mels, encoder_dims, decoder_dims, lstm_dims,
dropout, speaker_embedding_size) dropout, speaker_embedding_size)
self.postnet = CBHG(postnet_K, n_mels, postnet_dims, self.postnet = CBHG(postnet_K, n_mels, postnet_dims,