fix iterable dataset (issue 9899) (#9952)

This commit is contained in:
Yu Shan authored and GitHub committed 2020-08-22 19:40:38 -07:00
1 parent 245c0a9e43
commit 5264f888e4
1 file changed
+1 -1
@@ -140,7 +140,7 @@ class DistributedTorchRunner(TorchRunner):
if self.add_dist_sampler:
self.train_loader = with_sampler(self.train_loader)
if self.validation_loader and should_wrap_dataloader(
if self.validation_loader is not None and should_wrap_dataloader(
self.validation_loader):
if self.add_dist_sampler:
self.validation_loader = with_sampler(self.validation_loader)