ref: inner train loop (intermediate step) 12/n (#3371)

* ref: inner train loop (intermediate step) 12/n * ref: inner train loop (intermediate step) 12/n * ref: inner train loop (intermediate step) 12/n
Lightning-AI · Sep 6, 2020 · d091faf · d091faf
1 parent 8eef97c
commit d091faf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pytorch_lightning/trainer/training_loop_temp.py b/pytorch_lightning/trainer/training_loop_temp.py
@@ -87,11 +87,10 @@ def on_train_epoch_start(self, epoch):
         model = self.trainer.get_model()
 
         # set seed for distributed sampler (enables shuffling for each epoch)
-        # TODO: move to accelerators
-        if (self.trainer.use_ddp or self.trainer.use_horovod or self.trainer.on_tpu) \
-                and hasattr(self.trainer.train_dataloader, 'sampler') \
-                and hasattr(self.trainer.train_dataloader.sampler, 'set_epoch'):
+        try:
             self.trainer.train_dataloader.sampler.set_epoch(epoch)
+        except Exception:
+            pass
 
         # update training progress in trainer and model
         model.current_epoch = epoch