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 `speedy_speecy_config.py` for the trainer

This commit is contained in:
Eren Gölge 2021-05-26 16:02:01 +02:00
parent 4e910993f1
commit 06ee57d816
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/speedy_speech_config.py
View File

@ -1,4 +1,5 @@
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import List
from TTS.tts.configs.shared_configs import BaseTTSConfig from TTS.tts.configs.shared_configs import BaseTTSConfig
@ -99,10 +100,11 @@ class SpeedySpeechConfig(BaseTTSConfig):
external_speaker_embedding_file: str = False external_speaker_embedding_file: str = False
# optimizer parameters # optimizer parameters
noam_schedule: bool = False optimizer: str = "RAdam"
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
# loss params # loss params
@ -114,3 +116,12 @@ class SpeedySpeechConfig(BaseTTSConfig):
min_seq_len: int = 13 min_seq_len: int = 13
max_seq_len: int = 200 max_seq_len: int = 200
r: int = 1 # DO NOT CHANGE r: int = 1 # DO NOT CHANGE
# 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."
])