From 15c11fc848d1b6ac0e077dbf04dc21a1778452b8 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 11 May 2020 09:47:33 -0400 Subject: [PATCH] fixes no val loader --- pytorch_lightning/callbacks/progress.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/callbacks/progress.py b/pytorch_lightning/callbacks/progress.py index a770c6c9..fa7d0d7d 100644 --- a/pytorch_lightning/callbacks/progress.py +++ b/pytorch_lightning/callbacks/progress.py @@ -93,7 +93,7 @@ class ProgressBarBase(Callback): """ trainer = self.trainer total_val_batches = 0 - if trainer.fast_dev_run: + if trainer.fast_dev_run and trainer.val_dataloaders is not None: total_val_batches = len(trainer.val_dataloaders) elif not self.trainer.disable_validation: is_val_epoch = (trainer.current_epoch + 1) % trainer.check_val_every_n_epoch == 0