Fix fast_dev_run running validation twice (Lightning-AI#1365)

tullie · May 6, 2020 · 7ee1726 · 7ee1726
1 parent 3ecda02
commit 7ee1726
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -331,10 +331,10 @@ def __init__(
 
         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