Update to xavier_uniform and avoid legacy data.uniform_ initialization (#2018)

Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/2018

Reviewed By: hwangjeff, mthrok

Differential Revision: D32540498

Pulled By: nateanl

fbshipit-source-id: f104fbf84c0f489906b6555f21f931b60fedb59e

torchaudio/models/tacotron2.py

@@ -26,7 +26,6 @@
 # *****************************************************************************
 
 import warnings
-from math import sqrt
 from typing import Tuple, List, Optional, Union
 
 import torch
@@ -984,9 +983,7 @@ class Tacotron2(nn.Module):
         self.n_mels = n_mels
         self.n_frames_per_step = n_frames_per_step
         self.embedding = nn.Embedding(n_symbol, symbol_embedding_dim)
-        std = sqrt(2.0 / (n_symbol + symbol_embedding_dim))
-        val = sqrt(3.0) * std
-        self.embedding.weight.data.uniform_(-val, val)
+        torch.nn.init.xavier_uniform_(self.embedding.weight)
         self.encoder = _Encoder(
             encoder_embedding_dim, encoder_n_convolution, encoder_kernel_size
         )