wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleaning up demos

Browse Source
This commit is contained in:
William Falcon
2019-10-05 15:45:41 -04:00
parent e7f15e4e89
commit 7b5a40b170
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+3 -3
View File
@@ -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 = """