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

Add paramter for eval meta file on compute embeddings script

This commit is contained in:
Edresson Casanova 2022-09-21 13:59:13 -03:00
parent d6ad9a05b4
commit 89c6e2a32f
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
TTS/bin/compute_embeddings.py
View File

@ -62,9 +62,15 @@ parser.add_argument(
default=None, default=None,
) )
parser.add_argument( parser.add_argument(
"--metafile", "--meta_file_train",
type=str, type=str,
help="Path to the meta file. If not set, dataset formatter uses the default metafile if it is defined in the formatter. You either need to provide this or `config_dataset_path`", help="Path to the train meta file. If not set, dataset formatter uses the default metafile if it is defined in the formatter. You either need to provide this or `config_dataset_path`",
default=None,
)
parser.add_argument(
"--meta_file_val",
type=str,
help="Path to the evaluation meta file. If not set, dataset formatter uses the default metafile if it is defined in the formatter. You either need to provide this or `config_dataset_path`",
default=None, default=None,
) )
args = parser.parse_args() args = parser.parse_args()
@ -79,7 +85,8 @@ else:
c_dataset.formatter = args.formatter_name c_dataset.formatter = args.formatter_name
c_dataset.dataset_name = args.dataset_name c_dataset.dataset_name = args.dataset_name
c_dataset.path = args.dataset_path c_dataset.path = args.dataset_path
c_dataset.meta_file_train = args.metafile if args.metafile else None c_dataset.meta_file_train = args.meta_file_train if args.meta_file_train else None
c_dataset.meta_file_val = args.meta_file_val if args.meta_file_val else None
meta_data_train, meta_data_eval = load_tts_samples(c_dataset, eval_split=not args.no_eval) meta_data_train, meta_data_eval = load_tts_samples(c_dataset, eval_split=not args.no_eval)