mirror of https://github.com/coqui-ai/TTS.git
Update trainer.py
This commit is contained in:
parent
f3e9d61330
commit
8a246cbb66
|
@ -812,7 +812,7 @@ class Trainer:
|
||||||
samples = self.eval_loader.dataset.load_test_samples(1)
|
samples = self.eval_loader.dataset.load_test_samples(1)
|
||||||
figures, audios = self.model.test_run(self.ap, samples, None)
|
figures, audios = self.model.test_run(self.ap, samples, None)
|
||||||
else:
|
else:
|
||||||
figures, audios = self.model.test_run()
|
figures, audios = self.model.test_run(self.ap)
|
||||||
self.dashboard_logger.test_audios(self.total_steps_done, audios, self.config.audio["sample_rate"])
|
self.dashboard_logger.test_audios(self.total_steps_done, audios, self.config.audio["sample_rate"])
|
||||||
self.dashboard_logger.test_figures(self.total_steps_done, figures)
|
self.dashboard_logger.test_figures(self.total_steps_done, figures)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue