Debug purpose alignment image saving

2019-01-09 16:01:23 +01:00 · 2019-01-09 16:01:23 +01:00 · 656542db93
parent c2637f114e
commit 656542db93
1 changed files with 4 additions and 0 deletions
--- 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,