Update progress.py
This commit is contained in:
parent
15c11fc848
commit
88c086bbd2
|
@ -93,7 +93,7 @@ class ProgressBarBase(Callback):
|
||||||
"""
|
"""
|
||||||
trainer = self.trainer
|
trainer = self.trainer
|
||||||
total_val_batches = 0
|
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)
|
total_val_batches = len(trainer.val_dataloaders)
|
||||||
elif not self.trainer.disable_validation:
|
elif not self.trainer.disable_validation:
|
||||||
is_val_epoch = (trainer.current_epoch + 1) % trainer.check_val_every_n_epoch == 0
|
is_val_epoch = (trainer.current_epoch + 1) % trainer.check_val_every_n_epoch == 0
|
||||||
|
|
Loading…
Reference in New Issue