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 FastPitch config

This commit is contained in:
Eren Gölge 2021-07-22 14:20:11 +02:00
parent 8584f2b82d
commit 7692bfe7f8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
TTS/tts/configs/fast_pitch_config.py
View File

@ -63,15 +63,15 @@ class FastPitchConfig(BaseTTSConfig):
d_vector_dim: int = 0
# optimizer parameters
optimizer: str = "RAdam"
optimizer_params: dict = field(default_factory=lambda: {"betas": [0.9, 0.998], "weight_decay": 1e-6})
optimizer: str = "Adam"
optimizer_params: dict = field(default_factory=lambda: {"betas": [0.9, 0.98], "weight_decay": 1e-6})
lr_scheduler: str = "NoamLR"
lr_scheduler_params: dict = field(default_factory=lambda: {"warmup_steps": 4000})
lr: float = 1e-4
grad_clip: float = 5.0
grad_clip: float = 1000.0
# loss params
ssim_loss_alpha: float = 1.0
ssim_loss_alpha: float = 0.0
dur_loss_alpha: float = 1.0
spec_loss_alpha: float = 1.0
pitch_loss_alpha: float = 1.0