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

set dp as default backend

Browse Source
This commit is contained in:
William Falcon
2019-07-18 11:58:27 -04:00
parent 4085b3fa69
commit 0d992689d5
1 changed files with 2 additions and 2 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
@@ -106,9 +106,9 @@ class LightningTemplateModel(LightningModule):
output = OrderedDict({
'val_loss': loss_val,
'val_acc': torch.tensor(val_acc),
'val_acc': torch.tensor(val_acc).cuda(loss_val.device.index),
})
return loss_val
return output
def validation_end(self, outputs):
"""