From 304d60197b3bdd94001bfa0fb9162c769440f392 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sat, 5 Jun 2021 11:48:16 +0200 Subject: [PATCH] reduce multiband melgan test model size --- tests/vocoder_tests/test_fullband_melgan_train.py | 1 - tests/vocoder_tests/test_multiband_melgan_train.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/vocoder_tests/test_fullband_melgan_train.py b/tests/vocoder_tests/test_fullband_melgan_train.py index fbce03eb..f93a5318 100644 --- a/tests/vocoder_tests/test_fullband_melgan_train.py +++ b/tests/vocoder_tests/test_fullband_melgan_train.py @@ -20,7 +20,6 @@ config = FullbandMelganConfig( eval_split_size=1, print_step=1, print_eval=True, - discriminator_model_params={"base_channels": 16, "max_channels": 256, "downsample_factors": [4, 4, 4]}, data_path="tests/data/ljspeech", discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]}, output_path=output_path, diff --git a/tests/vocoder_tests/test_multiband_melgan_train.py b/tests/vocoder_tests/test_multiband_melgan_train.py index ef362414..5c6a0fc8 100644 --- a/tests/vocoder_tests/test_multiband_melgan_train.py +++ b/tests/vocoder_tests/test_multiband_melgan_train.py @@ -22,6 +22,7 @@ config = MultibandMelganConfig( print_eval=True, discriminator_model_params={"base_channels": 16, "max_channels": 128, "downsample_factors": [4, 4, 4]}, data_path="tests/data/ljspeech", + discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]}, output_path=output_path, ) config.audio.do_trim_silence = True