Docs: Fix typos in README (#323)

Xilinx · May 24, 2021 · 2d65dcd · 2d65dcd
1 parent 9cf09a1
commit 2d65dcd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -95,11 +95,11 @@ class QuantWeightLeNet(Module):
         self.fc3   = qnn.QuantLinear(84, 10, bias=False, weight_bit_width=3)
 
     def forward(self, x):
-        out = self.relu1(self.conv1(out))
+        out = self.relu1(self.conv1(x))
         out = F.max_pool2d(out, 2)
         out = self.relu2(self.conv2(out))
         out = F.max_pool2d(out, 2)
-        out = out.reshape(out.reshape[0], -1)
+        out = out.reshape(out.shape[0], -1)
         out = self.relu3(self.fc1(out))
         out = self.relu4(self.fc2(out))
         out = self.fc3(out)