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

Remove useless config argument

This commit is contained in:
Eren Golge 2018-03-28 09:43:29 -07:00
parent 75dd1bf2ad
commit f6f1b06b77
3 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
config.json
View File

@ -7,7 +7,6 @@
"preemphasis": 0.97, "preemphasis": 0.97,
"min_level_db": -100, "min_level_db": -100,
"ref_level_db": 20, "ref_level_db": 20,
"hidden_size": 128,
"embedding_size": 256, "embedding_size": 256,
"text_cleaner": "english_cleaners", "text_cleaner": "english_cleaners",

4
models/tacotron.py
View File

@ -8,7 +8,7 @@ from TTS.layers.tacotron import Prenet, Encoder, Decoder, CBHG
class Tacotron(nn.Module): class Tacotron(nn.Module):
def __init__(self, embedding_dim=256, linear_dim=1025, mel_dim=80, def __init__(self, embedding_dim=256, linear_dim=1025, mel_dim=80,
freq_dim=1025, r=5, padding_idx=None): r=5, padding_idx=None):
super(Tacotron, self).__init__() super(Tacotron, self).__init__()
self.r = r self.r = r
@ -24,7 +24,7 @@ class Tacotron(nn.Module):
self.decoder = Decoder(256, mel_dim, r) self.decoder = Decoder(256, mel_dim, r)
self.postnet = CBHG(mel_dim, K=8, projections=[256, mel_dim]) self.postnet = CBHG(mel_dim, K=8, projections=[256, mel_dim])
self.last_linear = nn.Linear(mel_dim * 2, freq_dim) self.last_linear = nn.Linear(mel_dim * 2, linear_dim)
def forward(self, characters, mel_specs=None): def forward(self, characters, mel_specs=None):

3
train.py
View File

@ -332,9 +332,8 @@ def main(args):
pin_memory=True) pin_memory=True)
model = Tacotron(c.embedding_size, model = Tacotron(c.embedding_size,
c.hidden_size,
c.num_mels,
c.num_freq, c.num_freq,
c.num_mels,
c.r) c.r)
optimizer = optim.Adam(model.parameters(), lr=c.lr) optimizer = optim.Adam(model.parameters(), lr=c.lr)