From d60a8d721110bdf27c66222223c48b21ffe79bb6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 15 Apr 2021 15:30:06 +0200 Subject: [PATCH] show the real waveform on TB too for GAN vocoder training. --- TTS/bin/train_vocoder_gan.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/TTS/bin/train_vocoder_gan.py b/TTS/bin/train_vocoder_gan.py index 43b1ff35..38aa69e5 100644 --- a/TTS/bin/train_vocoder_gan.py +++ b/TTS/bin/train_vocoder_gan.py @@ -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)