Fix train_log name

This commit is contained in:
Jindrich Matousek 2022-08-09 11:00:06 +02:00
parent 61508bf336
commit af2aee5ba9
1 changed files with 1 additions and 1 deletions

View File

@ -209,7 +209,7 @@ class GAN(BaseVocoder):
self, batch: Dict, outputs: Dict, logger: "Logger", assets: Dict, steps: int # pylint: disable=unused-argument self, batch: Dict, outputs: Dict, logger: "Logger", assets: Dict, steps: int # pylint: disable=unused-argument
) -> Tuple[Dict, np.ndarray]: ) -> Tuple[Dict, np.ndarray]:
"""Call `_log()` for training.""" """Call `_log()` for training."""
figures, audios = self._log("eval", self.ap, batch, outputs) figures, audios = self._log("train", self.ap, batch, outputs)
logger.eval_figures(steps, figures) logger.eval_figures(steps, figures)
logger.eval_audios(steps, audios, self.ap.sample_rate) logger.eval_audios(steps, audios, self.ap.sample_rate)