From 67de4c241e0b5dcde88957ecc9def408625e0432 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 15:14:34 -0400 Subject: [PATCH] cleaning up demos --- examples/basic_examples/lightning_module_template.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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