From 7b5a40b170fef7f55ca497df805a9889dde36f28 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 15:45:41 -0400 Subject: [PATCH] cleaning up demos --- pytorch_lightning/trainer/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 0b6a18f6..7476c8ff 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -639,7 +639,7 @@ class Trainer(TrainerIO): # 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): + if self.use_ddp or self.use_ddp2 and not isinstance(self.get_train_dataloader().sampler, DistributedSampler): msg = """ You're using multiple gpus and multiple nodes without using a DistributedSampler to assign a subset of your data to each process. To silence this warning, pass a @@ -658,7 +658,7 @@ class Trainer(TrainerIO): """ warnings.warn(msg) - if self.use_ddp and self.get_val_dataloaders is not None: + if self.use_ddp or self.use_ddp2 and self.get_val_dataloaders is not None: for dataloader in self.get_val_dataloaders(): if not isinstance(dataloader.sampler, DistributedSampler): msg = """ @@ -681,7 +681,7 @@ class Trainer(TrainerIO): warnings.warn(msg) break - if self.use_ddp and self.get_test_dataloaders is not None: + if self.use_ddp or self.use_ddp2 and self.get_test_dataloaders is not None: for dataloader in self.get_test_dataloaders(): if not isinstance(dataloader.sampler, DistributedSampler): msg = """