From 4b96bfe92568fbf5e25d9b0bfa595537ee319c92 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sat, 5 Feb 2022 20:30:55 +0100 Subject: [PATCH] Fix train logging --- TTS/tts/models/vits.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 751187ea..7dac1bb9 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -947,8 +947,8 @@ class Vits(BaseTTS): Tuple[Dict, np.ndarray]: training plots and output waveform. """ figures, audios = self._log(self.ap, batch, outputs, "train") - logger.eval_figures(steps, figures) - logger.eval_audios(steps, audios, self.ap.sample_rate) + logger.train_figures(steps, figures) + logger.train_figures(steps, audios, self.ap.sample_rate) @torch.no_grad() def eval_step(self, batch: dict, criterion: nn.Module, optimizer_idx: int):