updated args

2019-06-25 20:20:12 -04:00 · 2019-06-25 20:20:12 -04:00 · 9b46f13230
parent 078bbc5df5
commit 9b46f13230
1 changed files with 1 additions and 1 deletions
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@ -97,7 +97,7 @@ class ExampleModel(RootModule):

        # output = {'y_hat': y_hat, 'val_loss': loss_val.item(), 'val_acc': val_acc}
        output = OrderedDict({
-            'loss_val': loss_val,
+            'val_loss': loss_val,
            'val_acc': torch.tensor(val_acc),
        })
        return output