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

fix argument order for testing models

This commit is contained in:
Eren Golge 2019-07-19 11:49:03 +02:00
parent 4336e1d338
commit 85bff627c9
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_tacotron2_model.py
View File

@ -38,7 +38,7 @@ class TacotronTrainTest(unittest.TestCase):
criterion = MSELossMasked().to(device) criterion = MSELossMasked().to(device)
criterion_st = nn.BCEWithLogitsLoss().to(device) criterion_st = nn.BCEWithLogitsLoss().to(device)
model = Tacotron2(24, c.r, 5).to(device) model = Tacotron2(num_chars=24, r=c.r, num_speakers=5).to(device)
model.train() model.train()
model_ref = copy.deepcopy(model) model_ref = copy.deepcopy(model)
count = 0 count = 0

4
tests/test_tacotron_model.py
View File

@ -45,8 +45,8 @@ class TacotronTrainTest(unittest.TestCase):
criterion = L1LossMasked().to(device) criterion = L1LossMasked().to(device)
criterion_st = nn.BCEWithLogitsLoss().to(device) criterion_st = nn.BCEWithLogitsLoss().to(device)
model = Tacotron( model = Tacotron(
32, num_chars=32,
5, num_speakers=5,
linear_dim=c.audio['num_freq'], linear_dim=c.audio['num_freq'],
mel_dim=c.audio['num_mels'], mel_dim=c.audio['num_mels'],
r=c.r, r=c.r,