From d4dbd897521b6a62de15a3b59ffba209631d5c8d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 8 Jun 2021 17:34:19 +0200 Subject: [PATCH] fix calculation of `loader_start_time` --- TTS/tts/trainer_tts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/trainer_tts.py b/TTS/tts/trainer_tts.py index 9d060498..6c900120 100644 --- a/TTS/tts/trainer_tts.py +++ b/TTS/tts/trainer_tts.py @@ -476,8 +476,8 @@ class TrainerTTS(TrainerAbstract): else: batch_num_steps = int(len(self.train_loader.dataset) / self.config.batch_size) self.c_logger.print_train_start() - loader_start_time = time.time() for cur_step, batch in enumerate(self.train_loader): + loader_start_time = time.time() _, _ = self.train_step(batch, batch_num_steps, cur_step, loader_start_time) epoch_time = time.time() - epoch_start_time # Plot self.epochs_done Stats