malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 01:32:35 +00:00
Code Issues Releases Wiki Activity

model.py: setting weight initialization to xavier uniform

Browse Source
This commit is contained in:
rafaelvalle
2018-06-07 20:28:52 -07:00
parent d10da5f41e
commit 12ab5ba89c
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model.py
View File

@@ -459,6 +459,8 @@ class Tacotron2(nn.Module):
self.n_frames_per_step = hparams.n_frames_per_step self.n_frames_per_step = hparams.n_frames_per_step
self.embedding = nn.Embedding( self.embedding = nn.Embedding(
hparams.n_symbols, hparams.symbols_embedding_dim) hparams.n_symbols, hparams.symbols_embedding_dim)
torch.nn.init.xavier_uniform_(self.embedding.weight.data)
self.encoder = Encoder(hparams) self.encoder = Encoder(hparams)
self.decoder = Decoder(hparams) self.decoder = Decoder(hparams)
self.postnet = Postnet(hparams) self.postnet = Postnet(hparams)