wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixing logic (#1734)

Browse Source
This commit is contained in:
Tian Wang
2020-05-05 14:07:26 -04:00
committed by GitHub
parent 2a2f303ae9
commit d6a0375974
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/training_loop.py
+1 -1
View File
@@ -325,7 +325,7 @@ class TrainerTrainLoopMixin(ABC):
if self.reload_dataloaders_every_epoch:
self.reset_train_dataloader(model)
# set seed for distributed sampler (enables shuffling for each epoch)
if self.use_ddp or self.use_horovod \
if (self.use_ddp or self.use_horovod) \
and hasattr(self.train_dataloader.sampler, 'set_epoch'):
self.train_dataloader.sampler.set_epoch(epoch)