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

update model size to paper

This commit is contained in:
Eren 2018-09-06 14:36:53 +02:00
parent 8e564be3c7
commit 7d66bdc5f4
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
layers/tacotron.py
View File

@ -249,10 +249,10 @@ class PostCBHG(nn.Module):
self.cbhg = CBHG( self.cbhg = CBHG(
mel_dim, mel_dim,
K=8, K=8,
conv_bank_features=80, conv_bank_features=128,
conv_projections=[160, mel_dim], conv_projections=[256, mel_dim],
highway_features=80, highway_features=128,
gru_features=80, gru_features=128,
num_highways=4) num_highways=4)
def forward(self, x): def forward(self, x):
return self.cbhg(x) return self.cbhg(x)

2
train.py
View File

@ -401,7 +401,7 @@ def main(args):
criterion.cuda() criterion.cuda()
criterion_st.cuda() criterion_st.cuda()
optimizer.load_state_dict(checkpoint['optimizer']) optimizer.load_state_dict(checkpoint['optimizer'])
optimizer_st.load_state_dict(checkpoint['optimizer_st']) # optimizer_st.load_state_dict(checkpoint['optimizer_st'])
for state in optimizer.state.values(): for state in optimizer.state.values():
for k, v in state.items(): for k, v in state.items():
if torch.is_tensor(v): if torch.is_tensor(v):