From 85bff627c955b4d8f3efe788a02d755ac080b124 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 19 Jul 2019 11:49:03 +0200 Subject: [PATCH] fix argument order for testing models --- tests/test_tacotron2_model.py | 2 +- tests/test_tacotron_model.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_tacotron2_model.py b/tests/test_tacotron2_model.py index c2c39231..bda83a03 100644 --- a/tests/test_tacotron2_model.py +++ b/tests/test_tacotron2_model.py @@ -38,7 +38,7 @@ class TacotronTrainTest(unittest.TestCase): criterion = MSELossMasked().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_ref = copy.deepcopy(model) count = 0 diff --git a/tests/test_tacotron_model.py b/tests/test_tacotron_model.py index a80b2080..7b343423 100644 --- a/tests/test_tacotron_model.py +++ b/tests/test_tacotron_model.py @@ -45,8 +45,8 @@ class TacotronTrainTest(unittest.TestCase): criterion = L1LossMasked().to(device) criterion_st = nn.BCEWithLogitsLoss().to(device) model = Tacotron( - 32, - 5, + num_chars=32, + num_speakers=5, linear_dim=c.audio['num_freq'], mel_dim=c.audio['num_mels'], r=c.r,