From 2ac5cce67a550d1bb5fa332ec2a3b92a6ad17869 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 20:09:40 -0400 Subject: [PATCH] updated args --- docs/source/examples/example_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py index fa796f84..2cc2d5a0 100644 --- a/docs/source/examples/example_model.py +++ b/docs/source/examples/example_model.py @@ -65,6 +65,7 @@ class ExampleModel(RootModule): """ # forward pass x, y = data_batch + print(x.shape) x = x.view(x.size(0), -1) y_hat = self.forward(x) @@ -86,6 +87,7 @@ class ExampleModel(RootModule): :return: """ x, y = data_batch + print(x.shape) x = x.view(x.size(0), -1) y_hat = self.forward(x)