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

moved sampler

Browse Source
This commit is contained in:
William Falcon
2019-07-08 18:59:16 -04:00
parent 31d9062b3a
commit cf4b25e455
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/lightning_module_template.py
+2 -1
View File
@@ -173,10 +173,11 @@ class LightningTemplateModel(LightningModule):
# print('no sampler')
# train_sampler = None
should_shuffle = train_sampler is None
loader = DataLoader(
dataset=dataset,
batch_size=self.hparams.batch_size,
shuffle=True,
shuffle=should_shuffle,
sampler=train_sampler
)