Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 12:49:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

update glow_tts_config.py for setting the optimizer and the scheduler

Browse Source
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
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