cleaning up demos

2026-09-10 12:21:57 +08:00 · 2019-10-05 15:14:34 -04:00
parent a7e154e643
commit 67de4c241e
1 changed files with 2 additions and 1 deletions
@@ -202,7 +202,8 @@ class LightningTemplateModel(LightningModule):
            dataset=dataset,
            batch_size=batch_size,
            shuffle=should_shuffle,
-            sampler=train_sampler
+            sampler=train_sampler,
+            num_workers=0
        )

        return loader