moved sampler

2019-07-08 18:59:16 -04:00 · 2019-07-08 18:59:16 -04:00 · cf4b25e455
parent 31d9062b3a
commit cf4b25e455
1 changed files with 2 additions and 1 deletions
--- a/examples/new_project_templates/lightning_module_template.py
+++ b/examples/new_project_templates/lightning_module_template.py
@ -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
        )