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 docstring to compute_embeddings script

This commit is contained in:
Edresson Casanova 2021-07-21 07:16:10 -03:00 committed by GitHub
parent b1620d1f3f
commit d5adc35fdf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/bin/compute_embeddings.py
View File

@ -4,13 +4,18 @@ import os
from tqdm import tqdm from tqdm import tqdm
from argparse import RawTextHelpFormatter
from TTS.config import load_config from TTS.config import load_config
from TTS.tts.datasets import load_meta_data from TTS.tts.datasets import load_meta_data
from TTS.tts.utils.speakers import SpeakerManager from TTS.tts.utils.speakers import SpeakerManager
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description='Compute embedding vectors for each wav file in a dataset.' description="""Compute embedding vectors for each wav file in a dataset.\n\n"""
"""
Example runs:
python TTS/bin/compute_embeddings.py speaker_encoder_model.pth.tar speaker_encoder_config.json dataset_config.json embeddings_output_path/
""",
formatter_class=RawTextHelpFormatter,
) )
parser.add_argument("model_path", type=str, help="Path to model checkpoint file.") parser.add_argument("model_path", type=str, help="Path to model checkpoint file.")
parser.add_argument( parser.add_argument(