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