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

bug fix

This commit is contained in:
root 2020-01-27 16:28:47 +01:00
parent f4678cbd6b
commit 9d669d1024
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_tacotron_model.py
View File

@ -44,7 +44,7 @@ class TacotronTrainTest(unittest.TestCase):
stop_targets = (stop_targets.sum(2) > stop_targets = (stop_targets.sum(2) >
0.0).unsqueeze(2).float().squeeze() 0.0).unsqueeze(2).float().squeeze()
criterion = L1LossMasked().to(device) criterion = L1LossMasked(seq_len_norm=False).to(device)
criterion_st = nn.BCEWithLogitsLoss().to(device) criterion_st = nn.BCEWithLogitsLoss().to(device)
model = Tacotron( model = Tacotron(
num_chars=32, num_chars=32,