cleaning up demos

This commit is contained in:
William Falcon
2019-10-05 15:36:12 -04:00
parent 6be5fb3cf7
commit c0c1cf058f
+2
View File
@@ -636,6 +636,8 @@ class Trainer(TrainerIO):
self.get_test_dataloaders = model.test_dataloader
self.get_val_dataloaders = model.val_dataloader
# 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):
msg = """