unused_speakers argument for ignoring speaker ids in multi-speaker

training
This commit is contained in:
Eren Gölge 2021-05-18 14:50:03 +02:00
parent c2c7dff805
commit 4df31f7fbd
1 changed files with 5 additions and 4 deletions

View File

@ -143,9 +143,11 @@ class BaseDatasetConfig(Coqpit):
Dataset name that defines the preprocessor in use. Defaults to None.
path (str):
Root path to the dataset files. Defaults to None.
meta_file_train (Union[str, List]):
meta_file_train (str):
Name of the dataset meta file. Or a list of speakers to be ignored at training for multi-speaker datasets.
Defaults to None.
unused_speakers (List):
List of speakers IDs that are not used at the training. Default None.
meta_file_val (str):
Name of the dataset meta file that defines the instances used at validation.
meta_file_attn_mask (str):
@ -155,9 +157,8 @@ class BaseDatasetConfig(Coqpit):
name: str = ""
path: str = ""
meta_file_train: Union[
str, List
] = "" # TODO: don't take ignored speakers for multi-speaker datasets over this. This is Union for SC-Glow compat.
meta_file_train: str = ""
ununsed_speakers: List[str] = None
meta_file_val: str = ""
meta_file_attn_mask: str = ""