coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

lint fix and eval as argparse in extract tts spectrograms

pull/581/head
Edresson 2021-07-13 02:15:31 -03:00
parent 2e5baffa9c
commit d906fea08c
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/bin/compute_embeddings.py
View File

@ -1,13 +1,10 @@
import argparse
import glob
import os
import torch
from tqdm import tqdm
from TTS.config import BaseDatasetConfig, load_config
from TTS.speaker_encoder.utils.generic_utils import setup_model
from TTS.config import load_config
from TTS.tts.datasets import load_meta_data
from TTS.tts.utils.speakers import SpeakerManager

3
TTS/bin/extract_tts_spectrograms.py
View File

@ -227,7 +227,7 @@ def main(args): # pylint: disable=redefined-outer-name
ap = AudioProcessor(**c.audio)
# load data instances
meta_data_train, meta_data_eval = load_meta_data(c.datasets, eval_split=True, ignore_generated_eval=True)
meta_data_train, meta_data_eval = load_meta_data(c.datasets, eval_split=args.eval, ignore_generated_eval=True)
# use eval and training partitions
meta_data = meta_data_train + meta_data_eval
@ -271,6 +271,7 @@ if __name__ == "__main__":
parser.add_argument("--debug", default=False, action="store_true", help="Save audio files for debug")
parser.add_argument("--save_audio", default=False, action="store_true", help="Save audio files")
parser.add_argument("--quantized", action="store_true", help="Save quantized audio files")
parser.add_argument("--eval", type=bool, help="compute eval.", default=True)
args = parser.parse_args()
c = load_config(args.config_path)