reduce fullband melgan model size for testing

2021-06-02 11:43:27 +02:00 · 2021-06-02 11:43:27 +02:00 · db48c69f0f
parent 4baa59d732
commit db48c69f0f
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
@ -20,6 +20,7 @@ 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",
    output_path=output_path,
 )