From 438a4f74d0b9fd12647062fd87c7ca78c8bc0cf2 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 23:46:09 -0400 Subject: [PATCH] cleaned up test_tube logger --- pytorch_lightning/trainer/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index c767028b..18cd9781 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -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 = """