From 9d669d1024e554d3460963c7a31a94810dd2f442 Mon Sep 17 00:00:00 2001 From: root Date: Mon, 27 Jan 2020 16:28:47 +0100 Subject: [PATCH] bug fix --- tests/test_tacotron_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_tacotron_model.py b/tests/test_tacotron_model.py index 48627697..ac6712b0 100644 --- a/tests/test_tacotron_model.py +++ b/tests/test_tacotron_model.py @@ -44,7 +44,7 @@ class TacotronTrainTest(unittest.TestCase): stop_targets = (stop_targets.sum(2) > 0.0).unsqueeze(2).float().squeeze() - criterion = L1LossMasked().to(device) + criterion = L1LossMasked(seq_len_norm=False).to(device) criterion_st = nn.BCEWithLogitsLoss().to(device) model = Tacotron( num_chars=32,