We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2a2f303 commit d6a0375Copy full SHA for d6a0375
pytorch_lightning/trainer/training_loop.py
@@ -325,7 +325,7 @@ def train(self):
325
if self.reload_dataloaders_every_epoch:
326
self.reset_train_dataloader(model)
327
# set seed for distributed sampler (enables shuffling for each epoch)
328
- if self.use_ddp or self.use_horovod \
+ if (self.use_ddp or self.use_horovod) \
329
and hasattr(self.train_dataloader.sampler, 'set_epoch'):
330
self.train_dataloader.sampler.set_epoch(epoch)
331
0 commit comments