wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleaning up demos

Browse Source
This commit is contained in:
William Falcon
2019-10-05 15:08:39 -04:00
parent f2fb219bfe
commit a7e154e643
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/basic_examples/lightning_module_template.py
+2 -4
View File
@@ -195,8 +195,7 @@ class LightningTemplateModel(LightningModule):
batch_size = self.hparams.batch_size
if self.use_ddp:
train_sampler = DistributedSampler(dataset, rank=self.trainer.proc_rank)
batch_size = batch_size // self.trainer.world_size # scale batch size
train_sampler = DistributedSampler(dataset)
should_shuffle = train_sampler is None
loader = DataLoader(
@@ -249,6 +248,5 @@ class LightningTemplateModel(LightningModule):
# training params (opt)
parser.add_argument('--optimizer_name', default='adam', type=str)
parser.add_argument('--batch_size', default=256, type=int,
help='batch size will be divided over all gpus being used across all nodes')
parser.add_argument('--batch_size', default=64, type=int)
return parser