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

added multiple outputs to LightningTestModel

Browse Source
This commit is contained in:
William Falcon
2019-07-24 17:23:19 -04:00
parent 516ee9c985
commit 63ce8af27c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/testing_models/lm_test_module.py
+3 -3
View File
@@ -127,16 +127,16 @@ class LightningTestModel(LightningModule):
val_acc = val_acc.unsqueeze(0)
# alternate possible outputs to test
if self.trainer.batch_nb % 0 == 0:
if self.trainer.batch_nb % 1 == 0:
output = OrderedDict({
'val_loss': loss_val,
'val_acc': val_acc,
})
return output
if self.trainer.batch_nb % 1 == 0:
if self.trainer.batch_nb % 2 == 0:
return val_acc
if self.trainer.batch_nb % 2 == 0:
if self.trainer.batch_nb % 3 == 0:
output = OrderedDict({
'val_loss': loss_val,
'val_acc': val_acc,