mirror of
https://github.com/wassname/TTS.git
synced 2026-10-27 12:30:27 +08:00
bug fix
This commit is contained in:
1 parent
3ad38621d4
commit
7d2ef7fbbc
1 file changed
+1
-1
@@ -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