updated args

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