From 7d2ef7fbbce132c4de1374b1e087db1813f8f448 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 14 Dec 2018 11:02:58 +0100 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 37b48cba..783cf5b8 100644 --- a/train.py +++ b/train.py @@ -193,7 +193,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, figures = {"prediction": plot_spectrogram(const_spec, ap), "ground_truth": plot_spectrogram(gt_spec, ap), "alignment": plot_alignment(align_img)} - tb_logger.tb_train_figures(figures, current_step) + tb_logger.tb_train_figures(current_step, figures) # Sample audio tb_logger.tb_train_audios(current_step,