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)
|
||||
figures, audios = self.model.test_run(self.ap, samples, None)
|
||||
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_figures(self.total_steps_done, figures)
|
||||
|
||||
|
|
Loading…
Reference in New Issue