From d5adc35fdfd87a417431a549358711220ad971e1 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Wed, 21 Jul 2021 07:16:10 -0300 Subject: [PATCH] Add docstring to compute_embeddings script --- TTS/bin/compute_embeddings.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/TTS/bin/compute_embeddings.py b/TTS/bin/compute_embeddings.py index 7ea1e4f9..f485514a 100644 --- a/TTS/bin/compute_embeddings.py +++ b/TTS/bin/compute_embeddings.py @@ -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(