diff --git a/examples/basic_examples/lightning_module_template.py b/examples/basic_examples/lightning_module_template.py index f35ec4e5..61caa830 100644 --- a/examples/basic_examples/lightning_module_template.py +++ b/examples/basic_examples/lightning_module_template.py @@ -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