plot eval alignment score right

2020-12-30 03:04:58 +01:00 · 2020-12-30 03:04:58 +01:00 · 8971c59b2d
parent 3fa408a5ea
commit 8971c59b2d
1 changed files with 1 additions and 1 deletions
--- a/TTS/bin/train_speedy_speech.py
+++ b/TTS/bin/train_speedy_speech.py
@ -317,7 +317,7 @@ def evaluate(data_loader, model, criterion, ap, global_step, epoch):
            epoch_time += step_time

            # compute alignment score
-            align_error = 1 - alignment_diagonal_score(alignments)
+            align_error = 1 - alignment_diagonal_score(alignments, binary=True)
            loss_dict['align_error'] = align_error

            # aggregate losses from processes