Update 2. 改造后的mnist_train.ipynb

577092689 · Aug 31, 2017 · e480af0 · e480af0
1 parent e8c6c92
commit e480af0
Showing 1 changed file with 10 additions and 17 deletions.
diff --git a/Deep_Learning_with_TensorFlow/1.0.0/Chapter09/2. 改造后的mnist_train.ipynb b/Deep_Learning_with_TensorFlow/1.0.0/Chapter09/2. 改造后的mnist_train.ipynb
@@ -3,9 +3,7 @@
   {
    "cell_type": "code",
    "execution_count": 1,
-   "metadata": {
-    "collapsed": false
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import tensorflow as tf\n",
@@ -46,9 +44,7 @@
   {
    "cell_type": "code",
    "execution_count": 3,
-   "metadata": {
-    "collapsed": false
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "def train(mnist):\n",
@@ -83,11 +79,12 @@
     "\n",
     "        with tf.control_dependencies([train_step, variables_averages_op]):\n",
     "            train_op = tf.no_op(name='train')\n",
-    "            \n",
+    "    \n",
+    "    writer = tf.summary.FileWriter(\"/log/modified_mnist_train.log\", tf.get_default_graph())\n",
+    "    \n",
     "    # 训练模型。\n",
     "    with tf.Session() as sess:\n",
     "        tf.global_variables_initializer().run()\n",
-    "        writer = tf.summary.FileWriter(\"/log/modified_mnist_train.log\", tf.get_default_graph())\n",
     "        for i in range(TRAINING_STEPS):\n",
     "            xs, ys = mnist.train.next_batch(BATCH_SIZE)\n",
     "\n",
@@ -99,14 +96,12 @@
     "                _, loss_value, step = sess.run(\n",
     "                    [train_op, loss, global_step], feed_dict={x: xs, y_: ys},\n",
     "                    options=run_options, run_metadata=run_metadata)\n",
-    "                writer.add_run_metadata(run_metadata=run_metadata, tag=(\"tag%d\"%i), global_step=i),\n",
+    "                writer.add_run_metadata(run_metadata=run_metadata, tag=(\"tag%d\" % i), global_step=i)\n",
     "                print(\"After %d training step(s), loss on training batch is %g.\" % (step, loss_value))\n",
     "            else:\n",
     "                _, loss_value, step = sess.run([train_op, loss, global_step], feed_dict={x: xs, y_: ys})\n",
-    "\n",
-    "\n",
-    "    
-    "        writer.close()"
+    "                \n",
+    "    writer.close()"
    ]
   },
   {
@@ -119,9 +114,7 @@
   {
    "cell_type": "code",
    "execution_count": 4,
-   "metadata": {
-    "collapsed": false
-   },
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
@@ -163,7 +156,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython2",
-   "version": "2.7.10"
+   "version": "2.7.13"
   }
  },
  "nbformat": 4,