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

cleaned up test_tube logger

Browse Source
This commit is contained in:
William Falcon
2019-10-05 23:46:09 -04:00
parent d0a9c92326
commit 438a4f74d0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+2 -2
View File
@@ -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 = """