Update progress.py

This commit is contained in:
William Falcon 2020-05-11 09:48:15 -04:00 committed by GitHub
parent 15c11fc848
commit 88c086bbd2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -93,7 +93,7 @@ class ProgressBarBase(Callback):
"""
trainer = self.trainer
total_val_batches = 0
if trainer.fast_dev_run and trainer.val_dataloaders is not None:
if trainer.fast_dev_run:
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