bug fix and check cleaner config field by comparing with the list of

avail functions
This commit is contained in:
erogol 2020-03-17 15:05:13 +01:00
Родитель fa795347a9
Коммит b9df54adcd
2 изменённых файлов: 4 добавлений и 2 удалений

Просмотреть файл

@ -496,7 +496,8 @@ def check_config(c):
_check_argument('tb_model_param_stats', c, restricted=True, val_type=bool) _check_argument('tb_model_param_stats', c, restricted=True, val_type=bool)
# dataloading # dataloading
_check_argument('text_cleaner', c, restricted=True, val_type=str, enum_list=['english_cleaners', 'phoneme_cleaners', 'transliteration_cleaners', 'basic_cleaners']) from TTS.utils.text import cleaners
_check_argument('text_cleaner', c, restricted=True, val_type=str, enum_list=dir(cleaners))
_check_argument('enable_eos_bos_chars', c, restricted=True, val_type=bool) _check_argument('enable_eos_bos_chars', c, restricted=True, val_type=bool)
_check_argument('num_loader_workers', c, restricted=True, val_type=int, min_val=0) _check_argument('num_loader_workers', c, restricted=True, val_type=int, min_val=0)
_check_argument('num_val_loader_workers', c, restricted=True, val_type=int, min_val=0) _check_argument('num_val_loader_workers', c, restricted=True, val_type=int, min_val=0)
@ -518,4 +519,4 @@ def check_config(c):
_check_argument('name', dataset_entry, restricted=True, val_type=str) _check_argument('name', dataset_entry, restricted=True, val_type=str)
_check_argument('path', dataset_entry, restricted=True, val_type=str) _check_argument('path', dataset_entry, restricted=True, val_type=str)
_check_argument('meta_file_train', dataset_entry, restricted=True, val_type=str) _check_argument('meta_file_train', dataset_entry, restricted=True, val_type=str)
_check_argument('meta_file_val', dataset_entry, restricted=True, val_type=str) _check_argument('meta_file_val', dataset_entry, restricted=True, val_type=str)

Просмотреть файл

@ -97,6 +97,7 @@ def basic_turkish_cleaners(text):
text = text.replace("I", "ı") text = text.replace("I", "ı")
text = lowercase(text) text = lowercase(text)
text = collapse_whitespace(text) text = collapse_whitespace(text)
return text
def english_cleaners(text): def english_cleaners(text):