wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

enable fast_dev_run without a validation loop (#1779)

Browse Source 
* fix val dataloader

* Update evaluation_loop.py
This commit is contained in:
William Falcon
2020-05-11 11:30:22 -04:00
committed by GitHub
parent 88c086bbd2
commit eeb411144f
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/callbacks/progress.py
+1 -1
View File
@@ -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