From 275c759993900e60015492ec020ccd201f64f919 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 23 Dec 2021 15:57:10 +0000 Subject: [PATCH] Fix #1037 --- TTS/vocoder/models/gan.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py index e36c2cd1..76fee505 100644 --- a/TTS/vocoder/models/gan.py +++ b/TTS/vocoder/models/gan.py @@ -202,7 +202,9 @@ class GAN(BaseVocoder): ) -> Tuple[Dict, np.ndarray]: """Call `_log()` for training.""" ap = assets["audio_processor"] - self._log("train", ap, batch, outputs) + figures, audios = self._log("eval", ap, batch, outputs) + logger.eval_figures(steps, figures) + logger.eval_audios(steps, audios, ap.sample_rate) @torch.no_grad() def eval_step(self, batch: Dict, criterion: nn.Module, optimizer_idx: int) -> Tuple[Dict, Dict]: @@ -214,7 +216,9 @@ class GAN(BaseVocoder): ) -> Tuple[Dict, np.ndarray]: """Call `_log()` for evaluation.""" ap = assets["audio_processor"] - self._log("eval", ap, batch, outputs) + figures, audios = self._log("eval", ap, batch, outputs) + logger.eval_figures(steps, figures) + logger.eval_audios(steps, audios, ap.sample_rate) def load_checkpoint( self,