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

---
 docs/source/examples/example_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py
index 3238dab7..5bf1ead7 100644
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@@ -75,7 +75,7 @@ class ExampleModel(RootModule):
         # return loss_val, tqdm_dic
 
         output = OrderedDict({
-            'loss_val': loss_val,
+            'tng_loss': loss_val,
         })
         return output