From 45708d4a28e4aed26a54da824d80858f94c4809f Mon Sep 17 00:00:00 2001 From: Tullie Murrell Date: Fri, 3 Apr 2020 21:00:26 +0200 Subject: [PATCH] Fix fast_dev_run running validation twice (#1365) --- pytorch_lightning/trainer/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index e346f1f499700b..81c5bae574d769 100644 --- 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