wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

show the real waveform on TB too for GAN vocoder training.

Browse Source
This commit is contained in:
Eren Gölge
2021-04-15 15:30:06 +02:00
parent 5fbe926429
commit d60a8d7211
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/bin/train_vocoder_gan.py
+3 -2
View File
@@ -443,8 +443,9 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
tb_logger.tb_eval_figures(global_step, figures)
# Sample audio
sample_voice = y_hat[0].squeeze(0).detach().cpu().numpy()
tb_logger.tb_eval_audios(global_step, {"eval/audio": sample_voice}, c.audio["sample_rate"])
predict_waveform = y_hat[0].squeeze(0).detach().cpu().numpy()
real_waveform = y_G[0].squeeze(0).cpu().numpy()
tb_logger.tb_eval_audios(global_step, {"eval/audio": predict_waveform, "eval/real_waveformo": real_waveform}, c.audio["sample_rate"])
tb_logger.tb_eval_stats(global_step, keep_avg.avg_values)