From c90fce8d8c98fc58d9ab329a2ca95bc2539d4061 Mon Sep 17 00:00:00 2001 From: thllwg Date: Fri, 14 Aug 2020 11:06:32 +0200 Subject: [PATCH] Fix for #497 --- mozilla_voice_tts/bin/train_tts.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mozilla_voice_tts/bin/train_tts.py b/mozilla_voice_tts/bin/train_tts.py index 2b6cbfd0..9fef0888 100644 --- a/mozilla_voice_tts/bin/train_tts.py +++ b/mozilla_voice_tts/bin/train_tts.py @@ -100,7 +100,7 @@ def format_data(data, speaker_mapping=None): if c.use_speaker_embedding: if c.use_external_speaker_embedding_file: - speaker_embeddings = data[8] + speaker_embeddings = data[8] speaker_ids = None else: speaker_ids = [ @@ -624,7 +624,7 @@ def main(args): # pylint: disable=redefined-outer-name train_avg_loss_dict, global_step = train(model, criterion, optimizer, optimizer_st, scheduler, ap, global_step, epoch, amp, speaker_mapping) - eval_avg_loss_dict = evaluate(model, criterion, ap, global_step, epoch) + eval_avg_loss_dict = evaluate(model, criterion, ap, global_step, epoch, speaker_mapping) c_logger.print_epoch_end(epoch, eval_avg_loss_dict) target_loss = train_avg_loss_dict['avg_postnet_loss'] if c.run_eval: