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

Fix fast_dev_run running validation twice (#1365)

This commit is contained in:
Tullie Murrell 2020-04-03 21:00:26 +02:00 committed by GitHub
parent b31edf37bf
commit 38e89dd890
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/trainer/trainer.py
View File

@ -331,10 +331,10 @@ class Trainer(
self.fast_dev_run = fast_dev_run self.fast_dev_run = fast_dev_run
if self.fast_dev_run: if self.fast_dev_run:
self.num_sanity_val_steps = 1 self.num_sanity_val_steps = 0
self.max_epochs = 1 self.max_epochs = 1
log.info('Running in fast_dev_run mode: will run a full train,' log.info('Running in fast_dev_run mode: will run a full train,'
' val loop using a single batch') ' val and test loop using a single batch')
# set default save path if user didn't provide one # set default save path if user didn't provide one
self.default_save_path = default_save_path self.default_save_path = default_save_path