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

update aling_tts_config for the trainer

This commit is contained in:
Eren Gölge 2021-05-27 10:25:25 +02:00
parent e298b8e364
commit fc9a0fb8ce
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
TTS/tts/configs/align_tts_config.py
View File

@ -92,13 +92,24 @@ class AlignTTSConfig(BaseTTSConfig):
external_speaker_embedding_file: str = False external_speaker_embedding_file: str = False
# optimizer parameters # optimizer parameters
noam_schedule: bool = False optimizer: str = "Adam"
warmup_steps: int = 4000 optimizer_params: dict = field(default_factory=lambda: {'betas': [0.9, 0.998], 'weight_decay': 1e-6})
lr_scheduler: str = None
lr_scheduler_params: dict = None
lr: float = 1e-4 lr: float = 1e-4
wd: float = 1e-6
grad_clip: float = 5.0 grad_clip: float = 5.0
# overrides # overrides
min_seq_len: int = 13 min_seq_len: int = 13
max_seq_len: int = 200 max_seq_len: int = 200
r: int = 1 r: int = 1
# testing
test_sentences: List[str] = field(default_factory=lambda:[
"It took me quite a long time to develop a voice, and now that I have it I'm not going to be silent.",
"Be a voice, not an echo.",
"I'm sorry Dave. I'm afraid I can't do that.",
"This cake is great. It's so delicious and moist.",
"Prior to November 22, 1963."
])