This commit is contained in:
Eren Golge 2018-12-14 11:02:58 +01:00
parent 3ad38621d4
commit 7d2ef7fbbc
1 changed files with 1 additions and 1 deletions

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,