From 5dabf85a3cc54e98b93c1bc17fd7c7fe2ce28d39 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Wed, 1 Jul 2020 01:11:25 +0200 Subject: [PATCH] test fix --- pytorch_lightning/trainer/training_loop.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index 3ee497305304a..338f66cb9c955 100644 --- 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