mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 23:54:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update export-onnx.py

Browse Source
This commit is contained in:
jinzr 2023-12-05 08:53:20 +08:00
parent d4c830823c
commit e401a724ac
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/vctk/TTS/vits/export-onnx.py
View File

@ -68,6 +68,12 @@ def get_parser():
help="The experiment dir", help="The experiment dir",
) )
parser.add_argument(
"--speakers",
type=Path,
default=Path("data/speakers.txt"),
help="Path to speakers.txt file.",
)
parser.add_argument( parser.add_argument(
"--tokens", "--tokens",
type=str, type=str,
@ -219,6 +225,10 @@ def main():
params.oov_id = tokenizer.oov_id params.oov_id = tokenizer.oov_id
params.vocab_size = tokenizer.vocab_size params.vocab_size = tokenizer.vocab_size
with open(args.speakers) as f:
speaker_map = {line.strip(): i for i, line in enumerate(f)}
params.num_spks = len(speaker_map)
logging.info(params) logging.info(params)
logging.info("About to create model") logging.info("About to create model")