From 6c7bbcaef04a911c3f49c24eb50a55f691ea7194 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 25 Jun 2021 16:52:17 +0200 Subject: [PATCH] Use `en-us` for testing phoneme models --- tests/tts_tests/test_glow_tts_train.py | 3 ++- tests/tts_tests/test_speedy_speech_train.py | 2 +- tests/vocoder_tests/test_multiband_melgan_train.py | 1 + 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/tts_tests/test_glow_tts_train.py b/tests/tts_tests/test_glow_tts_train.py index 2e675d13..e44f6365 100644 --- a/tests/tts_tests/test_glow_tts_train.py +++ b/tests/tts_tests/test_glow_tts_train.py @@ -16,7 +16,8 @@ config = GlowTTSConfig( num_val_loader_workers=0, text_cleaner="english_cleaners", use_phonemes=True, - phoneme_language="zh-CN", + use_espeak_phonemes=True, + phoneme_language="en-us", phoneme_cache_path="tests/data/ljspeech/phoneme_cache/", run_eval=True, test_delay_epochs=-1, diff --git a/tests/tts_tests/test_speedy_speech_train.py b/tests/tts_tests/test_speedy_speech_train.py index 3f508117..9dcf0ad8 100644 --- a/tests/tts_tests/test_speedy_speech_train.py +++ b/tests/tts_tests/test_speedy_speech_train.py @@ -16,7 +16,7 @@ config = SpeedySpeechConfig( num_val_loader_workers=0, text_cleaner="english_cleaners", use_phonemes=True, - phoneme_language="zh-CN", + phoneme_language="en-us", phoneme_cache_path="tests/data/ljspeech/phoneme_cache/", run_eval=True, test_delay_epochs=-1, diff --git a/tests/vocoder_tests/test_multiband_melgan_train.py b/tests/vocoder_tests/test_multiband_melgan_train.py index 081fb40e..ef362414 100644 --- a/tests/vocoder_tests/test_multiband_melgan_train.py +++ b/tests/vocoder_tests/test_multiband_melgan_train.py @@ -20,6 +20,7 @@ config = MultibandMelganConfig( eval_split_size=1, print_step=1, print_eval=True, + discriminator_model_params={"base_channels": 16, "max_channels": 128, "downsample_factors": [4, 4, 4]}, data_path="tests/data/ljspeech", output_path=output_path, )