diff --git a/02_MNIST_SLN_solution.ipynb b/02_MNIST_SLN_solution.ipynb index 8717e35..1538366 100644 --- a/02_MNIST_SLN_solution.ipynb +++ b/02_MNIST_SLN_solution.ipynb @@ -295,8 +295,8 @@ "# and mean square error cost\n", "cost = MSECost()\n", "model = MLP(cost=cost)\n", - "model.add_layer(Linear(idim=784, odim=100, rng=rng))\n", - "model.add_layer(Linear(idim=100, odim=10, rng=rng))\n", + "model.add_layer(Linear(idim=784, odim=200, rng=rng))\n", + "model.add_layer(Linear(idim=200, odim=10, rng=rng))\n", "\n", "# define the optimiser, here stochasitc gradient descent\n", "# with fixed learning rate and max_epochs as stopping criterion\n",