From f02856c0bc66e8b2d012670f1a2308c4630a5835 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 15:15:28 -0400 Subject: [PATCH] cleaning up demos --- examples/basic_examples/lightning_module_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/basic_examples/lightning_module_template.py b/examples/basic_examples/lightning_module_template.py index 61caa830..5cb2c3ee 100644 --- a/examples/basic_examples/lightning_module_template.py +++ b/examples/basic_examples/lightning_module_template.py @@ -190,7 +190,7 @@ class LightningTemplateModel(LightningModule): dataset = MNIST(root=self.hparams.data_root, train=train, transform=transform, download=True) - # when using multi-node (ddp) we need to add the datasampler + # when using multi-node (ddp) we need to add the datasampler train_sampler = None batch_size = self.hparams.batch_size