From 8a246cbb6607deda8772a569f9a66f51fa9a718a Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Tue, 20 Jul 2021 20:26:26 +0530 Subject: [PATCH] Update trainer.py --- TTS/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/trainer.py b/TTS/trainer.py index 49f6a58c..d05028e2 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -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)