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

update `glow_tts_config.py` for setting the optimizer and the scheduler

pull/506/head
Eren Gölge 2021-05-26 09:52:29 +02:00
parent 891631ab47
commit b2218e882a
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/tts/configs/glow_tts_config.py
View File

@ -89,12 +89,13 @@ class GlowTTSConfig(BaseTTSConfig):
use_external_speaker_embedding_file: bool = False
external_speaker_embedding_file: str = False
# optimizer params
noam_schedule: bool = True
warmup_steps: int = 4000
# optimizer parameters
optimizer: str = "RAdam"
optimizer_params: dict = field(default_factory=lambda: {'betas': [0.9, 0.998], 'weight_decay': 1e-6})
lr_scheduler: str = "NoamLR"
lr_scheduler_params: dict = field(default_factory=lambda:{"warmup_steps": 4000})
grad_clip: float = 5.0
lr: float = 1e-3
wd: float = 0.000001
# overrides
min_seq_len: int = 3