From d5d47eab0d45a4207cd9dd6795ddf7685df2db82 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 23 Aug 2019 02:39:05 -0400 Subject: [PATCH] Update lightning_module_template.py --- examples/new_project_templates/lightning_module_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index adc81f44..b7349523 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -194,7 +194,7 @@ class LightningTemplateModel(LightningModule): batch_size = self.hparams.batch_size try: - if self.on_gpu: + if self.trainer.use_ddp: train_sampler = DistributedSampler(dataset, rank=self.trainer.proc_rank) batch_size = batch_size // self.trainer.world_size # scale batch size except Exception: