wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-10 11:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Get the number speaker from the Speaker Manager property

Browse Source
This commit is contained in:
Edresson
2021-12-20 11:54:10 +00:00
committed by Eren Gölge
parent f394d60695
commit 4196a42de7
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/bin/train_tts.py
+4 -5
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