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
pytorch_lightning/testing/lm_test_module.py
+1 -1
View File
@@ -231,7 +231,7 @@ class LightningTestModel(LightningModule):
@data_loader
def val_dataloader(self):
return self.__dataloader(train=False)
return [self.__dataloader(train=False), self.__dataloader(train=False)]
@data_loader
def test_dataloader(self):
pytorch_lightning/testing/no_val_end_module.py
+4 -4
View File
@@ -109,7 +109,7 @@ class NoValEndTestModel(LightningModule):
if self.trainer.batch_nb % 2 == 0:
return loss_val
def validation_step(self, data_batch, batch_i, dataloader_i):
def validation_step(self, data_batch, batch_nb):
"""
Lightning calls this inside the validation loop
:param data_batch:
@@ -135,16 +135,16 @@ class NoValEndTestModel(LightningModule):
val_acc = val_acc.unsqueeze(0)
# alternate possible outputs to test
if batch_i % 1 == 0:
if batch_nb % 1 == 0:
output = OrderedDict({
'val_loss': loss_val,
'val_acc': val_acc,
})
return output
if batch_i % 2 == 0:
if batch_nb % 2 == 0:
return val_acc
if batch_i % 3 == 0:
if batch_nb % 3 == 0:
output = OrderedDict({
'val_loss': loss_val,
'val_acc': val_acc,