Skip to content

Commit d6a0375

Browse files
authored
Fixing logic (#1734)
1 parent 2a2f303 commit d6a0375

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/trainer/training_loop.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ def train(self):
325325
if self.reload_dataloaders_every_epoch:
326326
self.reset_train_dataloader(model)
327327
# set seed for distributed sampler (enables shuffling for each epoch)
328-
if self.use_ddp or self.use_horovod \
328+
if (self.use_ddp or self.use_horovod) \
329329
and hasattr(self.train_dataloader.sampler, 'set_epoch'):
330330
self.train_dataloader.sampler.set_epoch(epoch)
331331

0 commit comments

Comments
 (0)