From e81dbce38ced1d67599fe972d99ca6806a50f5fd Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 18 Jul 2019 11:59:14 -0400
Subject: [PATCH] set dp as default backend

---
 examples/new_project_templates/lightning_module_template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py
index 7987591f..2371ed92 100644
--- a/examples/new_project_templates/lightning_module_template.py
+++ b/examples/new_project_templates/lightning_module_template.py
@@ -86,7 +86,7 @@ class LightningTemplateModel(LightningModule):
         output = OrderedDict({
             'loss': loss_val
         })
-        return loss_val
+        return output
 
     def validation_step(self, data_batch, batch_i):
         """