mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
bug fix
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user