test fix

Lightning-AI · Jun 30, 2020 · 5dabf85 · 5dabf85
1 parent 145670f
commit 5dabf85
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py
@@ -567,12 +567,15 @@ def save_loggers_in_training_loop(self, batch_idx):
 
     def should_check_val(self, batch_idx, is_last_batch):
         # decide if we should run validation
-        is_val_check_batch = (batch_idx + 1) % self.val_check_batch == 0
+        # print(batch_idx)
+        # print(self.val_check_batch)
+        is_val_check_batch = (batch_idx + 1 % self.val_check_batch) == 0
+        if is_val_check_batch:
+            print('VAAAAAAAA')
         can_check_epoch = (self.current_epoch + 1) % self.check_val_every_n_epoch == 0
         can_check_val = not self.disable_validation and can_check_epoch
         should_check_val = is_val_check_batch or self.should_stop
-        is_last_batch_for_infinite_dataset = (is_last_batch and self.val_check_batch == float('inf'))
-        should_check_val = can_check_val and (should_check_val or is_last_batch_for_infinite_dataset)
+        should_check_val = can_check_val and (should_check_val or is_last_batch)
 
         return should_check_val