fix: prevent iteration increment in validation

dbstj1231 · May 24, 2021 · 5af10c2 · 5af10c2
1 parent 20d8b3b
commit 5af10c2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)