Update neural_network.ipynb (aymericdamien#361)

Add the missing fully connected layer 2 in the RNN example
ppxing456 · May 16, 2020 · 2cf9bfd · 2cf9bfd
1 parent 39d9d0e
commit 2cf9bfd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network.ipynb b/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network.ipynb
@@ -116,6 +116,7 @@
     "    # Set forward pass.\n",
     "    def call(self, x, is_training=False):\n",
     "        x = self.fc1(x)\n",
+    "        x = self.fc2(x)\n",
     "        x = self.out(x)\n",
     "        if not is_training:\n",
     "            # tf cross entropy expect logits without softmax, so only\n",