Kikirilkov
commited on
Commit
•
97f62f8
1
Parent(s):
6088cb1
Update TTS/vocoder/configs/shared_configs.py
Browse files
TTS/vocoder/configs/shared_configs.py
CHANGED
@@ -50,7 +50,7 @@ class BaseVocoderConfig(BaseTrainingConfig):
|
|
50 |
conv_pad: int = 0 # additional padding against convolutions applied to spectrograms
|
51 |
use_cache: bool = False # use in memory cache to keep the computed features. This might cause OOM.
|
52 |
# OPTIMIZER
|
53 |
-
epochs: int =
|
54 |
wd: float = 0.0 # Weight decay weight.
|
55 |
optimizer: str = "AdamW"
|
56 |
optimizer_params: dict = field(default_factory=lambda: {"betas": [0.8, 0.99], "weight_decay": 0.0})
|
|
|
50 |
conv_pad: int = 0 # additional padding against convolutions applied to spectrograms
|
51 |
use_cache: bool = False # use in memory cache to keep the computed features. This might cause OOM.
|
52 |
# OPTIMIZER
|
53 |
+
epochs: int = 500 # total number of epochs to train.
|
54 |
wd: float = 0.0 # Weight decay weight.
|
55 |
optimizer: str = "AdamW"
|
56 |
optimizer_params: dict = field(default_factory=lambda: {"betas": [0.8, 0.99], "weight_decay": 0.0})
|