Fix fast_dev_run running validation twice (#1365)

This commit is contained in:
Tullie Murrell
2020-04-03 15:00:26 -04:00
committed by GitHub
parent b31edf37bf
commit 38e89dd890
+2 -2
View File
@@ -331,10 +331,10 @@ class Trainer(
self.fast_dev_run = fast_dev_run
if self.fast_dev_run:
self.num_sanity_val_steps = 1
self.num_sanity_val_steps = 0
self.max_epochs = 1
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
self.default_save_path = default_save_path