From 0bd9152e0a6bc780353f1b3bd4c8e1c850c82fe9 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 19:55:26 -0400 Subject: [PATCH] scaled batch size --- examples/new_project_templates/lightning_module_template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 826b642562..2eb73ba9a7 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -168,6 +168,7 @@ class LightningTemplateModel(LightningModule): print(dist.get_world_size()) train_sampler = DistributedSampler(dataset, rank=self.trainer.proc_rank) batch_size = batch_size // self.trainer.world_size # scale batch size + print(batch_size) # except Exception as e: # pass