From 5af10c28dd888ebfc8c63fc36d66c6d660dafe54 Mon Sep 17 00:00:00 2001 From: Jake Tae Date: Tue, 25 May 2021 02:18:12 +0900 Subject: [PATCH] fix: prevent iteration increment in validation --- trainer/core.py | 1 - 1 file changed, 1 deletion(-) diff --git a/trainer/core.py b/trainer/core.py index 37055f3..45dcd19 100644 --- a/trainer/core.py +++ b/trainer/core.py @@ -41,7 +41,6 @@ def _valid_epoch(self): self.model.eval() for batch in tqdm(self.valid_loader, leave=False): - self.iteration += 1 pitch, text, mel = to_device(batch, self.device) pred = self.model(pitch, text)