From 614738cc85ca0cffa29431611fa1ee4463e43909 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sun, 6 Jun 2021 13:42:52 +0200 Subject: [PATCH] downsize melgan test model size --- TTS/tts/datasets/formatters.py | 1 - tests/vocoder_tests/test_melgan_train.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/TTS/tts/datasets/formatters.py b/TTS/tts/datasets/formatters.py index 3cb37168..db7841f4 100644 --- a/TTS/tts/datasets/formatters.py +++ b/TTS/tts/datasets/formatters.py @@ -112,7 +112,6 @@ def ljspeech_test(root_path, meta_file): https://keithito.com/LJ-Speech-Dataset/""" txt_file = os.path.join(root_path, meta_file) items = [] - speaker_name = "ljspeech" with open(txt_file, "r", encoding="utf-8") as ttf: for idx, line in enumerate(ttf): cols = line.split("|") diff --git a/tests/vocoder_tests/test_melgan_train.py b/tests/vocoder_tests/test_melgan_train.py index 3ff65b5a..551b786a 100644 --- a/tests/vocoder_tests/test_melgan_train.py +++ b/tests/vocoder_tests/test_melgan_train.py @@ -19,7 +19,7 @@ config = MelganConfig( seq_len=2048, eval_split_size=1, print_step=1, - discriminator_model_params={"base_channels": 16, "max_channels": 256, "downsample_factors": [4, 4, 4]}, + discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]}, print_eval=True, data_path="tests/data/ljspeech", output_path=output_path,