From 4196a42de76f753ead8ca5c408844a933014d463 Mon Sep 17 00:00:00 2001 From: Edresson Date: Mon, 22 Nov 2021 18:52:59 -0300 Subject: [PATCH] Get the number speaker from the Speaker Manager property --- TTS/bin/train_tts.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/TTS/bin/train_tts.py b/TTS/bin/train_tts.py index f39ed259..a543a947 100644 --- a/TTS/bin/train_tts.py +++ b/TTS/bin/train_tts.py @@ -48,16 +48,15 @@ def main(): if config.use_speaker_embedding: speaker_manager = SpeakerManager(data_items=train_samples + eval_samples) if hasattr(config, "model_args"): - config.model_args.num_speakers = len(speaker_manager.speaker_ids) + config.model_args.num_speakers = speaker_manager.num_speakers else: - config.num_speakers = len(speaker_manager.speaker_ids) - + config.num_speakers = speaker_manager.num_speakers elif config.use_d_vector_file: speaker_manager = SpeakerManager(d_vectors_file_path=config.d_vector_file) if hasattr(config, "model_args"): - config.model_args.num_speakers = len(speaker_manager.speaker_ids) + config.model_args.num_speakers = speaker_manager.num_speakers else: - config.num_speakers = len(speaker_manager.speaker_ids) + config.num_speakers = speaker_manager.num_speakers else: speaker_manager = None