From 28cfddbe65897e94edcce827c1f54f51460292ab Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 16 Jul 2019 12:44:58 -0400
Subject: [PATCH] accept dist sampler classes

---
 pytorch_lightning/models/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 27fbf704..12031132 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -277,7 +277,7 @@ class Trainer(TrainerIO):
         self.test_dataloader = model.test_dataloader
         self.val_dataloader = model.val_dataloader
 
-        if self.data_parallel and type(self.tng_dataloader.sampler) is not DistributedSampler:
+        if self.data_parallel and issubclass(self.tng_dataloader.sampler, DistributedSampler):
             msg = '''
             when using multiple gpus and multiple nodes you must pass a DistributedSampler to DataLoader(sampler).