This commit is contained in:
Eren Golge
2018-12-28 14:18:33 +01:00
parent a1fe867371
commit 6d8b2dad91
+1 -1
View File
@@ -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,