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

View File

@ -17,7 +17,7 @@ class VitsConfig(BaseTTSConfig):
Model architecture arguments. Defaults to `VitsArgs()`.
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):
Initial learning rate for the generator. Defaults to 0.0002.
@ -114,7 +114,6 @@ class VitsConfig(BaseTTSConfig):
feat_loss_alpha: float = 1.0
mel_loss_alpha: float = 45.0
dur_loss_alpha: float = 1.0
aligner_loss_alpha = 1.0
speaker_encoder_loss_alpha: float = 1.0
# data loader params