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

Get the number speaker from the Speaker Manager property

pull/1032/head
Edresson 2021-11-22 18:52:59 -03:00 committed by Eren Gölge
parent f394d60695
commit 4196a42de7
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/bin/train_tts.py
View File

@ -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