From 0245a2e25891b16e3be11630ec0d74459f9c5768 Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 9 Jun 2020 23:01:34 +0200 Subject: [PATCH] change alignment_score to alignment_error --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index a5f3d2e7..869557e6 100644 --- a/train.py +++ b/train.py @@ -291,7 +291,7 @@ def train(model, criterion, optimizer, optimizer_st, scheduler, "loss_postnet": keep_avg['avg_postnet_loss'], "loss_decoder": keep_avg['avg_decoder_loss'], "stopnet_loss": keep_avg['avg_stopnet_loss'], - "alignment_score": keep_avg['avg_align_error'], + "alignment_error": keep_avg['avg_align_error'], "epoch_time": epoch_time } if c.ga_alpha > 0: @@ -413,7 +413,7 @@ def evaluate(model, criterion, ap, global_step, epoch): "loss_postnet": keep_avg['avg_postnet_loss'], "loss_decoder": keep_avg['avg_decoder_loss'], "stopnet_loss": keep_avg['avg_stopnet_loss'], - "alignment_score": keep_avg['avg_align_error'], + "alignment_error": keep_avg['avg_align_error'], } if c.bidirectional_decoder: