From c0c1cf058f288c00008579e1e03aad979fe8642d Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 15:36:12 -0400 Subject: [PATCH] cleaning up demos --- pytorch_lightning/trainer/trainer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 6b66e5ec..650af0fd 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -636,6 +636,8 @@ class Trainer(TrainerIO): self.get_test_dataloaders = model.test_dataloader self.get_val_dataloaders = model.val_dataloader + # call warnings from proc zero only which triggers dataloaders + # if those have to download data it will only happen on proc 0 if self.proc_rank == 0: if self.use_ddp and not isinstance(self.get_train_dataloader().sampler, DistributedSampler): msg = """