Spaces:
Running
Running
Kikirilkov
commited on
Commit
•
6258c41
1
Parent(s):
84f6279
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})
|
@@ -90,7 +90,7 @@ class BaseGANVocoderConfig(BaseVocoderConfig):
|
|
90 |
Parameters for the L1 spectrogram loss. Defaults to
|
91 |
`{
|
92 |
"use_mel": True,
|
93 |
-
"sample_rate":
|
94 |
"n_fft": 1024,
|
95 |
"hop_length": 256,
|
96 |
"win_length": 1024,
|
@@ -155,7 +155,7 @@ class BaseGANVocoderConfig(BaseVocoderConfig):
|
|
155 |
l1_spec_loss_params: dict = field(
|
156 |
default_factory=lambda: {
|
157 |
"use_mel": True,
|
158 |
-
"sample_rate":
|
159 |
"n_fft": 1024,
|
160 |
"hop_length": 256,
|
161 |
"win_length": 1024,
|
|
|
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 = 5000 # 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})
|
|
|
90 |
Parameters for the L1 spectrogram loss. Defaults to
|
91 |
`{
|
92 |
"use_mel": True,
|
93 |
+
"sample_rate": 24000,
|
94 |
"n_fft": 1024,
|
95 |
"hop_length": 256,
|
96 |
"win_length": 1024,
|
|
|
155 |
l1_spec_loss_params: dict = field(
|
156 |
default_factory=lambda: {
|
157 |
"use_mel": True,
|
158 |
+
"sample_rate": 24000,
|
159 |
"n_fft": 1024,
|
160 |
"hop_length": 256,
|
161 |
"win_length": 1024,
|