From 9ce41264825340666801082a59ed611eb718a9ee Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 17 Jul 2020 13:01:45 +0200 Subject: [PATCH] update pwgan generator test --- tests/test_vocoder_parallel_wavegan_generator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_vocoder_parallel_wavegan_generator.py b/tests/test_vocoder_parallel_wavegan_generator.py index f904ed24..55103ac4 100644 --- a/tests/test_vocoder_parallel_wavegan_generator.py +++ b/tests/test_vocoder_parallel_wavegan_generator.py @@ -22,9 +22,9 @@ def test_pwgan_generator(): use_causal_conv=False, upsample_conditional_features=True, upsample_factors=[4, 4, 4, 4]) - dummy_c = torch.rand((4, 80, 64)) + dummy_c = torch.rand((2, 80, 5)) output = model(dummy_c) - assert np.all(output.shape == (4, 1, 64 * 256)) + assert np.all(output.shape == (2, 1, 5 * 256)), output.shape model.remove_weight_norm() output = model.inference(dummy_c) - assert np.all(output.shape == (4, 1, (64 + 4) * 256)) + assert np.all(output.shape == (2, 1, (5 + 4) * 256))