Replace reduce_sum by reduce_mean for MSE (rather than SSE) error.

dezhili · Feb 2, 2017 · 3c6ad48 · 3c6ad48
1 parent cabe792
commit 3c6ad48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/15_autoencoders.ipynb b/15_autoencoders.ipynb
@@ -208,7 +208,7 @@
     "hidden = fully_connected(X, n_hidden, activation_fn=None)\n",
     "outputs = fully_connected(hidden, n_outputs, activation_fn=None)\n",
     "\n",
-    "mse = tf.reduce_sum(tf.square(outputs - X))\n",
+    "mse = tf.reduce_mean(tf.square(outputs - X))\n",
     "\n",
     "optimizer = tf.train.AdamOptimizer(learning_rate)\n",
     "training_op = optimizer.minimize(mse)\n",
@@ -2057,7 +2057,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.1"
+   "version": "3.5.2+"
   },
   "nav_menu": {
    "height": "381px",