model.py: setting weight initialization to xavier uniform

2018-06-07 20:28:52 -07:00 · 2018-06-07 20:28:52 -07:00 · 12ab5ba89c
parent d10da5f41e
commit 12ab5ba89c
1 changed files with 2 additions and 0 deletions
--- a/model.py
+++ b/model.py
@ -459,6 +459,8 @@ class Tacotron2(nn.Module):
        self.n_frames_per_step = hparams.n_frames_per_step
        self.embedding = nn.Embedding(
            hparams.n_symbols, hparams.symbols_embedding_dim)
+        torch.nn.init.xavier_uniform_(self.embedding.weight.data)
+
        self.encoder = Encoder(hparams)
        self.decoder = Decoder(hparams)
        self.postnet = Postnet(hparams)