mirror of https://github.com/coqui-ai/TTS.git
update `speedy_speecy_config.py` for the trainer
parent
c9790bee2c
commit
843b3ba960
|
@ -1,4 +1,5 @@
|
|||
from dataclasses import dataclass, field
|
||||
from typing import List
|
||||
|
||||
from TTS.tts.configs.shared_configs import BaseTTSConfig
|
||||
|
||||
|
@ -99,10 +100,11 @@ class SpeedySpeechConfig(BaseTTSConfig):
|
|||
external_speaker_embedding_file: str = False
|
||||
|
||||
# optimizer parameters
|
||||
noam_schedule: bool = False
|
||||
warmup_steps: int = 4000
|
||||
optimizer: str = "RAdam"
|
||||
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
|
||||
wd: float = 1e-6
|
||||
grad_clip: float = 5.0
|
||||
|
||||
# loss params
|
||||
|
@ -114,3 +116,12 @@ class SpeedySpeechConfig(BaseTTSConfig):
|
|||
min_seq_len: int = 13
|
||||
max_seq_len: int = 200
|
||||
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."
|
||||
])
|
||||
|
|
Loading…
Reference in New Issue