coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix unit tests issue

pull/1265/head
Edresson Casanova 2022-02-19 12:15:03 +00:00
parent fc7081fc5e
commit 5218d6b7a4
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/tts_tests/test_fast_pitch_speaker_emb_train.py
View File

@ -7,7 +7,7 @@ from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.fast_pitch_config import FastPitchConfig
from TTS.utils.trainer_utils import get_last_checkpoint
config_path = os.path.join(get_tests_output_path(), "test_fast_pitch_config.json")
config_path = os.path.join(get_tests_output_path(), "test_model_config.json")
output_path = os.path.join(get_tests_output_path(), "train_outputs")
audio_config = BaseAudioConfig(
@ -45,6 +45,8 @@ config = FastPitchConfig(
],
)
config.audio.do_trim_silence = True
config.use_speaker_embedding = True
config.model_args.use_speaker_embedding = True
config.audio.trim_db = 60
config.save_json(config_path)

6
tests/tts_tests/test_fast_pitch_train.py
View File

@ -7,7 +7,7 @@ from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.fast_pitch_config import FastPitchConfig
from TTS.utils.trainer_utils import get_last_checkpoint
config_path = os.path.join(get_tests_output_path(), "test_fast_pitch_config.json")
config_path = os.path.join(get_tests_output_path(), "test_model_config.json")
output_path = os.path.join(get_tests_output_path(), "train_outputs")
audio_config = BaseAudioConfig(
@ -42,8 +42,11 @@ config = FastPitchConfig(
test_sentences=[
"Be a voice, not an echo.",
],
use_speaker_embedding=False,
)
config.audio.do_trim_silence = True
config.use_speaker_embedding = False
config.model_args.use_speaker_embedding = False
config.audio.trim_db = 60
config.save_json(config_path)
@ -58,6 +61,7 @@ command_train = (
"--coqpit.datasets.0.meta_file_attn_mask tests/data/ljspeech/metadata_attn_mask.txt "
"--coqpit.test_delay_epochs 0"
)
run_cli(command_train)
# Find latest folder