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: