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 `glow_tts_config.py` for setting the optimizer and the scheduler

This commit is contained in:
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 use_external_speaker_embedding_file: bool = False
external_speaker_embedding_file: str = False external_speaker_embedding_file: str = False
# optimizer params # optimizer parameters
noam_schedule: bool = True 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 = "NoamLR"
lr_scheduler_params: dict = field(default_factory=lambda:{"warmup_steps": 4000})
grad_clip: float = 5.0 grad_clip: float = 5.0
lr: float = 1e-3 lr: float = 1e-3
wd: float = 0.000001
# overrides # overrides
min_seq_len: int = 3 min_seq_len: int = 3