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

cleaned up test_tube logger

Browse Source
This commit is contained in:
William Falcon
2019-10-05 23:35:10 -04:00
parent d0a2438e22
commit d0a9c92326
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 or self.use_ddp2 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 or self.use_ddp2 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 or self.use_ddp2 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 = """