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

View File

@ -4,13 +4,18 @@ import os
from tqdm import tqdm
from argparse import RawTextHelpFormatter
from TTS.config import load_config
from TTS.tts.datasets import load_meta_data
from TTS.tts.utils.speakers import SpeakerManager
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(