wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-26 13:00:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

scaled batch size

Browse Source
This commit is contained in:
William Falcon committed 2019-07-08 19:53:24 -04:00
1 parent b644234d08
commit f95fad864d
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/lightning_module_template.py
+1 -1
View File
@@ -164,7 +164,7 @@ class LightningTemplateModel(LightningModule):
# try:
# if self.on_gpu:
train_sampler = DistributedSampler(dataset, num_replicas=self.trainer.world_size, rank=self.trainer.proc_rank)
train_sampler = DistributedSampler(dataset, rank=self.trainer.proc_rank)
batch_size = batch_size // self.trainer.world_size # scale batch size
# except Exception as e: