reduce fullband-melgan test model size

2021-05-31 15:44:09 +02:00 · 2021-05-31 15:44:09 +02:00 · 6d6896fd99
parent 1443d03af1
commit 6d6896fd99
1 changed files with 1 additions and 0 deletions
--- a/tests/vocoder_tests/test_fullband_melgan_train.py
+++ b/tests/vocoder_tests/test_fullband_melgan_train.py
@ -22,6 +22,7 @@ config = FullbandMelganConfig(
    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": 128, "downsample_factors": [4, 4, 4]},
    output_path=output_path,
 )
 config.audio.do_trim_silence = True