Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix train logging

This commit is contained in:
Eren Gölge 2022-02-05 20:30:55 +01:00
parent ab8a4ca2c3
commit 4b96bfe925
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/vits.py
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):