From bac0ef2d4456dc3fa5f2592cfad84e085c984787 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 25 Jun 2019 20:08:32 -0400
Subject: [PATCH] updated args

---
 docs/source/examples/example_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py
index e7caed4bef..fa796f84eb 100644
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@@ -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):