mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
set dp as default backend
This commit is contained in:
1 parent
4085b3fa69
commit
0d992689d5
1 file changed
+2
-2
@@ -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):
|
||||
"""
|
||||
|
||||
Reference in new issue
Block a user