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

updated args

Browse Source
This commit is contained in:
William Falcon
2019-06-25 20:08:32 -04:00
parent d3b621dfd2
commit bac0ef2d44
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
+2 -2
View File
@@ -77,7 +77,7 @@ class ExampleModel(RootModule):
output = OrderedDict({
'loss_val': loss_val,
})
return output
return torch.tensor(4)
def validation_step(self, data_batch, batch_i):
"""
@@ -100,7 +100,7 @@ class ExampleModel(RootModule):
'loss_val': loss_val,
'val_acc': val_acc,
})
return output
return torch.tensor(4)
def validation_end(self, outputs):