From 656542db93249269be73e7818732ba617851e9af Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Wed, 9 Jan 2019 16:01:23 +0100
Subject: [PATCH] Debug purpose alignment image saving

---
 train.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/train.py b/train.py
index f52d365d..2be8071f 100644
--- a/train.py
+++ b/train.py
@@ -219,6 +219,10 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st,
                                     avg_stop_loss, epoch_time, avg_step_time),
         flush=True)
 
+    align_img = alignments[0].data.cpu().numpy()
+    align_img = plot_alignment(align_img)
+    align_img.savefig('/home/erogol/Desktop/alignment_{}.png'.format(current_step))
+
     # Plot Training Epoch Stats
     epoch_stats = {"loss_postnet": avg_linear_loss,
                    "loss_decoder": avg_mel_loss,