set dp as default backend

This commit is contained in:
William Falcon
2019-07-18 11:56:48 -04:00
parent f98f88ff08
commit b684bb55c5
@@ -86,7 +86,7 @@ class LightningTemplateModel(LightningModule):
output = OrderedDict({
'loss': loss_val
})
return output
return loss_val
def validation_step(self, data_batch, batch_i):
"""
@@ -108,7 +108,7 @@ class LightningTemplateModel(LightningModule):
'val_loss': loss_val,
'val_acc': torch.tensor(val_acc),
})
return output
return loss_val
def validation_end(self, outputs):
"""