reduce fullband-melgan test model size

2021-05-31 15:44:09 +02:00 · 2021-05-31 15:44:09 +02:00 · a58e986f68
parent 9960c0c356
commit a58e986f68
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