bug fix

2019-10-29 19:07:08 +01:00 · 2019-10-29 19:07:08 +01:00 · 8f53f9fc8f
parent 0e0d0345cd
commit 8f53f9fc8f
1 changed files with 4 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -283,8 +283,11 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
                    "prediction": plot_spectrogram(const_spec, ap),
                    "ground_truth": plot_spectrogram(gt_spec, ap),
                    "alignment": plot_alignment(align_img),
-                    "alignment_backward": plot_alignment(alignments_backward[0].data.cpu().numpy())
                }
+
+                if c.bidirectional_decoder:
+                    figures["alignment_backward"] = plot_alignment(alignments_backward[0].data.cpu().numpy())
+
                tb_logger.tb_train_figures(global_step, figures)

                # Sample audio