updated args

2019-06-25 20:08:32 -04:00 · 2019-06-25 20:08:32 -04:00 · bac0ef2d44
parent d3b621dfd2
commit bac0ef2d44
1 changed files with 2 additions and 2 deletions
--- 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):