coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix `test_run` for DDP

pull/725/head
Eren Gölge 2021-08-13 19:39:02 +00:00
parent 7c0d564965
commit c8bbcdfd07
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
TTS/trainer.py
View File

@ -829,9 +829,15 @@ class Trainer:
if hasattr(self.eval_loader.dataset, "load_test_samples"):
samples = self.eval_loader.dataset.load_test_samples(1)
figures, audios = self.model.test_run(self.ap, samples, None)
if self.num_gpus > 1:
figures, audios = self.model.module.test_run(self.ap, samples, None)
else:
figures, audios = self.model.test_run(self.ap, samples, None)
else:
figures, audios = self.model.test_run(self.ap)
if self.num_gpus > 1:
figures, audios = self.model.module.test_run(self.ap)
else:
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)