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

Update BaseTTS config

This commit is contained in:
Eren Gölge 2021-12-07 12:57:51 +00:00
parent 4c5cb44eeb
commit cfaa51fddc
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
TTS/tts/configs/shared_configs.py
View File

@ -78,7 +78,7 @@ class CharactersConfig(Coqpit):
is_unique (bool): is_unique (bool):
remove any duplicate characters in the character lists. It is a bandaid for compatibility with the old remove any duplicate characters in the character lists. It is a bandaid for compatibility with the old
models trained with character lists with duplicates. models trained with character lists with duplicates. Defaults to True.
is_sorted (bool): is_sorted (bool):
Sort the characters in alphabetical order. Defaults to True. Sort the characters in alphabetical order. Defaults to True.
@ -166,6 +166,9 @@ class BaseTTSConfig(BaseTrainingConfig):
compute_linear_spec (bool): compute_linear_spec (bool):
If True data loader computes and returns linear spectrograms alongside the other data. If True data loader computes and returns linear spectrograms alongside the other data.
precompute_num_workers (int):
Number of workers to precompute features. Defaults to 0.
use_noise_augment (bool): use_noise_augment (bool):
Augment the input audio with random noise. Augment the input audio with random noise.
@ -214,6 +217,7 @@ class BaseTTSConfig(BaseTrainingConfig):
phoneme_cache_path: str = None phoneme_cache_path: str = None
# vocabulary parameters # vocabulary parameters
characters: CharactersConfig = None characters: CharactersConfig = None
add_blank: bool = False
# training params # training params
batch_group_size: int = 0 batch_group_size: int = 0
loss_masking: bool = None loss_masking: bool = None
@ -225,8 +229,8 @@ class BaseTTSConfig(BaseTrainingConfig):
max_text_len: int = float("inf") max_text_len: int = float("inf")
compute_f0: bool = False compute_f0: bool = False
compute_linear_spec: bool = False compute_linear_spec: bool = False
precompute_num_workers: int = 0
use_noise_augment: bool = False use_noise_augment: bool = False
add_blank: bool = False
# dataset # dataset
datasets: List[BaseDatasetConfig] = field(default_factory=lambda: [BaseDatasetConfig()]) datasets: List[BaseDatasetConfig] = field(default_factory=lambda: [BaseDatasetConfig()])
# optimizer # optimizer