From 9d666af71529e699d2188dd07ad1ad6dc40a6aec Mon Sep 17 00:00:00 2001
From: root <root@sp-mlc3-5423-0.mlc>
Date: Fri, 3 Jan 2020 00:30:01 +0100
Subject: [PATCH] plot alignment score for eval

---
 train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 8aa0567d..d24a4b6b 100644
--- a/train.py
+++ b/train.py
@@ -455,7 +455,8 @@ def evaluate(model, criterion, criterion_st, ap, global_step, epoch):
                 epoch_stats = {
                     "loss_postnet": keep_avg['avg_postnet_loss'],
                     "loss_decoder": keep_avg['avg_decoder_loss'],
-                    "stop_loss": keep_avg['avg_stop_loss']
+                    "stop_loss": keep_avg['avg_stop_loss'],
+                    "alignment_score": keep_avg['avg_align_score']
                 }
 
                 if c.bidirectional_decoder: