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 = 10000 # 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})
 
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})