From 7285598e11b9b2512856a4ed26697b74d24628f3 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 19:57:51 -0400 Subject: [PATCH] scaled batch size --- examples/new_project_templates/lightning_module_template.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 359c85a8..994c1621 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -164,7 +164,6 @@ class LightningTemplateModel(LightningModule): try: if self.on_gpu: - print('distributing') train_sampler = DistributedSampler(dataset, rank=self.trainer.proc_rank) batch_size = batch_size // self.trainer.world_size # scale batch size except Exception as e: