update `speedy_speecy_config.py` for the trainer

pull/602/head
Eren Gölge 2021-05-26 16:02:01 +02:00
parent c9790bee2c
commit 843b3ba960
1 changed files with 14 additions and 3 deletions

View File

@ -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."
])