|
@@ -116,7 +116,7 @@
|
|
" # Set forward pass.\n",
|
|
" # Set forward pass.\n",
|
|
" def call(self, x, is_training=False):\n",
|
|
" def call(self, x, is_training=False):\n",
|
|
" x = self.fc1(x)\n",
|
|
" x = self.fc1(x)\n",
|
|
- " x = self.fc2(x)\n"
|
|
|
|
|
|
+ " x = self.fc2(x)\n",
|
|
" x = self.out(x)\n",
|
|
" x = self.out(x)\n",
|
|
" if not is_training:\n",
|
|
" if not is_training:\n",
|
|
" # tf cross entropy expect logits without softmax, so only\n",
|
|
" # tf cross entropy expect logits without softmax, so only\n",
|