Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use val_percent_check in validation step (#135)

This commit is contained in:
Maxim Andreev 2019-08-18 18:02:28 +03:00 committed by William Falcon
parent 9aa9a1a796
commit e646d745da
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch_lightning/models/trainer.py
View File

@ -1094,9 +1094,9 @@ If you want each process to load the full dataset, ignore this warning.
model = self.__get_model()
model.on_pre_performance_check()
# use full val set on end of epoch
# use val_percent_check set on end of epoch
# use a small portion otherwise
max_batches = None if not self.fast_dev_run else 1
max_batches = self.nb_val_batches if not self.fast_dev_run else 1
for ds_i, dataloader in enumerate(self.val_dataloader):
val_out_metrics = self.validate(self.model, dataloader, max_batches, ds_i)
self.__add_tqdm_metrics(val_out_metrics)