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)