wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix train logging

Browse Source
This commit is contained in:
Eren Gölge
2022-02-25 11:26:59 +01:00
parent ab8a4ca2c3
commit 4b96bfe925
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/models/vits.py
+2 -2
View File
@@ -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):