Update trainer.py

This commit is contained in:
Ayush Chaurasia 2021-07-20 20:26:26 +05:30 committed by Eren Gölge
parent f3e9d61330
commit 8a246cbb66
1 changed files with 1 additions and 1 deletions

View File

@ -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)