From 0738cb0efe11ba6df81430b1a75aae965789e6aa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Mon, 4 Apr 2022 09:46:10 +0200 Subject: [PATCH] Fix Vocoder logging --- TTS/vocoder/models/gan.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py index ed5b26dd..51cdefc2 100644 --- a/TTS/vocoder/models/gan.py +++ b/TTS/vocoder/models/gan.py @@ -209,9 +209,9 @@ class GAN(BaseVocoder): self, batch: Dict, outputs: Dict, logger: "Logger", assets: Dict, steps: int # pylint: disable=unused-argument ) -> Tuple[Dict, np.ndarray]: """Call `_log()` for training.""" - figures, audios = self._log("eval", self.ap, batch, outputs) - logger.eval_figures(steps, figures) - logger.eval_audios(steps, audios, self.ap.sample_rate) + figures, audios = self._log("train", self.ap, batch, outputs) + logger.train_figures(steps, figures) + logger.train_audios(steps, audios, self.ap.sample_rate) @torch.no_grad() def eval_step(self, batch: Dict, criterion: nn.Module, optimizer_idx: int) -> Tuple[Dict, Dict]: