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

Val idx optional in validation_step (#108)

Browse Source 
* made dataset_i only available with multiple datasets

* updated interface signature

* updated tests
This commit is contained in:
William Falcon
2019-08-13 11:37:37 -04:00
committed by GitHub
parent 905a2e5a12
commit 7f53e7bfb3
6 changed files with 48 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/lightning_module_template.py
+2 -2
View File
@@ -105,7 +105,7 @@ class LightningTemplateModel(LightningModule):
# can also return just a scalar instead of a dict (return loss_val)
return output
def validation_step(self, data_batch, batch_i, dataloader_i):
def validation_step(self, data_batch, batch_i):
"""
Lightning calls this inside the validation loop
:param data_batch:
@@ -218,7 +218,7 @@ class LightningTemplateModel(LightningModule):
@pl.data_loader
def val_dataloader(self):
print('val data loader called')
return [self.__dataloader(train=False) for i in range(2)]
return self.__dataloader(train=False)
@pl.data_loader
def test_dataloader(self):