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

Fix default values

This commit is contained in:
Eren Gölge 2022-02-03 15:37:30 +01:00
parent 54c6bb2a8c
commit d3a58ed07a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/tts/configs/vits_config.py
View File

@ -17,7 +17,7 @@ class VitsConfig(BaseTTSConfig):
Model architecture arguments. Defaults to `VitsArgs()`. Model architecture arguments. Defaults to `VitsArgs()`.
grad_clip (List): grad_clip (List):
Gradient clipping thresholds for each optimizer. Defaults to `[5.0, 5.0]`. Gradient clipping thresholds for each optimizer. Defaults to `[1000.0, 1000.0]`.
lr_gen (float): lr_gen (float):
Initial learning rate for the generator. Defaults to 0.0002. Initial learning rate for the generator. Defaults to 0.0002.
@ -114,7 +114,6 @@ class VitsConfig(BaseTTSConfig):
feat_loss_alpha: float = 1.0 feat_loss_alpha: float = 1.0
mel_loss_alpha: float = 45.0 mel_loss_alpha: float = 45.0
dur_loss_alpha: float = 1.0 dur_loss_alpha: float = 1.0
aligner_loss_alpha = 1.0
speaker_encoder_loss_alpha: float = 1.0 speaker_encoder_loss_alpha: float = 1.0
# data loader params # data loader params