Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change embedding layer init to old version

This commit is contained in:
Eren Golge 2019-01-15 15:51:55 +01:00
parent 5733eab90b
commit b78fc96115
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models/tacotron.py
View File

@ -21,9 +21,7 @@ class Tacotron(nn.Module):
self.embedding = nn.Embedding(
len(phonemes), embedding_dim, padding_idx=padding_idx)
print(" | > Number of characters : {}".format(len(phonemes)))
std = sqrt(2.0 / (len(phonemes) + embedding_dim))
val = sqrt(3.0) * std # uniform bounds for std
self.embedding.weight.data.uniform_(-val, val)
self.embedding.weight.data.normal_(0, 0.3)
self.encoder = Encoder(embedding_dim)
self.decoder = Decoder(256, mel_dim, r, attn_windowing)
self.postnet = PostCBHG(mel_dim)