updated args

2019-06-25 20:25:34 -04:00 · 2019-06-25 20:25:34 -04:00 · 7527167f69
parent 45331b396f
commit 7527167f69
1 changed files with 1 additions and 1 deletions
--- 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