diff --git a/notebooks/cae_mnist.ipynb b/notebooks/cae_mnist.ipynb index 0228b5d..0e32b10 100755 --- a/notebooks/cae_mnist.ipynb +++ b/notebooks/cae_mnist.ipynb @@ -100,7 +100,7 @@ " _ce2 = tf.nn.dropout(_ce2, _keepprob)\n", " _ce3 = tf.nn.sigmoid(tf.add(tf.nn.conv2d(_ce2, _W['ce3']\n", " , strides=[1, 2, 2, 1], padding='SAME'), _b['be3'])) \n", - " _ce1 = tf.nn.dropout(_ce3, _keepprob)\n", + " _ce3 = tf.nn.dropout(_ce3, _keepprob)\n", " # Decoder\n", " _cd3 = tf.nn.sigmoid(tf.add(tf.nn.conv2d_transpose(_ce3, _W['cd3']\n", " , tf.pack([tf.shape(_X)[0], 7, 7, n2]), strides=[1, 2, 2, 1]\n",