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

Merge pull request #2189 from coqui-ai/fix-capacitron-test

This commit is contained in:
Eren Gölge 2022-12-06 21:25:06 +01:00 committed by GitHub
parent 9321b22203 4787a2a993
commit c753ad49cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/tts_tests/test_tacotron2_model.py
View File

@ -301,7 +301,7 @@ class TacotronCapacitronTrainTest(unittest.TestCase):
batch["stop_targets"] = (batch["stop_targets"].sum(2) > 0.0).unsqueeze(2).float().squeeze() batch["stop_targets"] = (batch["stop_targets"].sum(2) > 0.0).unsqueeze(2).float().squeeze()
model = Tacotron2(config).to(device) model = Tacotron2(config).to(device)
criterion = model.get_criterion() criterion = model.get_criterion().to(device)
optimizer = model.get_optimizer() optimizer = model.get_optimizer()
model.train() model.train()