Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fix

This commit is contained in:
Eren Golge 2019-10-29 19:07:08 +01:00
parent 0e0d0345cd
commit 8f53f9fc8f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -283,8 +283,11 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
"prediction": plot_spectrogram(const_spec, ap), "prediction": plot_spectrogram(const_spec, ap),
"ground_truth": plot_spectrogram(gt_spec, ap), "ground_truth": plot_spectrogram(gt_spec, ap),
"alignment": plot_alignment(align_img), "alignment": plot_alignment(align_img),
"alignment_backward": plot_alignment(alignments_backward[0].data.cpu().numpy())
} }
if c.bidirectional_decoder:
figures["alignment_backward"] = plot_alignment(alignments_backward[0].data.cpu().numpy())
tb_logger.tb_train_figures(global_step, figures) tb_logger.tb_train_figures(global_step, figures)
# Sample audio # Sample audio