diff --git a/TTS/bin/train_speedy_speech.py b/TTS/bin/train_speedy_speech.py index d66ad268..6fd969b3 100644 --- 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