diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 3f073e726f7eb..6109c21b825d2 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -979,13 +979,9 @@ def on_train_end(self, trainer, pl_module): # check all validation data trainer = Trainer(num_sanity_val_steps=-1) -Example:: - python -m torch_xla.distributed.xla_dist - --tpu=$TPU_POD_NAME - --conda-env=torch-xla-nightly - --env=XLA_USE_BF16=1 - -- python your_trainer_file.py +This option will reset the validation dataloader unless ``num_sanity_val_steps=0``. + plugins ^^^^^^^