improve comments

Author: dsilver829

LeNet-Lab-Solution.ipynb

@@ -243,10 +243,10 @@
     "    # SOLUTION: Pooling. Input = 10x10x16. Output = 5x5x16.\n",
     "    conv2 = tf.nn.max_pool(conv2, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='VALID')\n",
     "\n",
-    "    # SOLUTION: Flatten.\n",
+    "    # SOLUTION: Flatten. Input = 5x5x16. Output = 400.\n",
     "    fc0   = flatten(conv2)\n",
     "    \n",
-    "    # SOLUTION: Layer 3: Fully Connected. Input = 400 (5x5x16). Output = 120.\n",
+    "    # SOLUTION: Layer 3: Fully Connected. Input = 400. Output = 120.\n",
     "    fc1_W = tf.Variable(tf.truncated_normal(shape=(400, 120), mean = mu, stddev = sigma))\n",
     "    fc1_b = tf.Variable(tf.zeros(120))\n",
     "    fc1   = tf.matmul(fc0, fc1_W) + fc1_b\n",
@@ -350,7 +350,7 @@
     "    sess = tf.get_default_session()\n",
     "    for offset in range(0, num_examples, BATCH_SIZE):\n",
     "        batch_x, batch_y = X_data[offset:offset+BATCH_SIZE], y_data[offset:offset+BATCH_SIZE]\n",
-    "        accuracy =  sess.run(accuracy_operation, feed_dict={x: batch_x, y: batch_y})\n",
+    "        accuracy = sess.run(accuracy_operation, feed_dict={x: batch_x, y: batch_y})\n",
     "        total_accuracy += (accuracy * len(batch_x))\n",
     "    return total_accuracy / num_examples"
    ]
@@ -390,7 +390,7 @@
     "        for offset in range(0, num_examples, BATCH_SIZE):\n",
     "            end = offset + BATCH_SIZE\n",
     "            batch_x, batch_y = X_train[offset:end], y_train[offset:end]\n",
-    "            loss = sess.run(training_operation, feed_dict={x: batch_x, y: batch_y})\n",
+    "            sess.run(training_operation, feed_dict={x: batch_x, y: batch_y})\n",
     "            \n",
     "        validation_accuracy = evaluate(X_validation, y_validation)\n",
     "        print(\"EPOCH {} ...\".format(i+1))\n",

LeNet-Lab.ipynb

@@ -233,9 +233,9 @@
     "\n",
     "    # TODO: Pooling. Input = 10x10x16. Output = 5x5x16.\n",
     "\n",
-    "    # TODO: Flatten.\n",
+    "    # TODO: Flatten. Input = 5x5x16. Output = 400.\n",
     "    \n",
-    "    # TODO: Layer 3: Fully Connected. Input = 400 (5x5x16). Output = 120.\n",
+    "    # TODO: Layer 3: Fully Connected. Input = 400. Output = 120.\n",
     "    \n",
     "    # TODO: Activation.\n",
     "\n",