Merge pull request nlintz#58 from gm06041/patch-7

hunkim · web-flow · commit a9f9a5b50075 · 2016-10-18T16:20:12.000+08:00
Update 10_save_restore_net.py
diff --git a/10_save_restore_net.py b/10_save_restore_net.py
@@ -77,6 +77,6 @@ def model(X, w_h, w_h2, w_o, p_keep_input, p_keep_hidden): # this network is the
         global_step.assign(i).eval() # set and update(eval) global_step with index, i
         saver.save(sess, ckpt_dir + "/model.ckpt", global_step=global_step)
         print(i, np.mean(np.argmax(teY, axis=1) ==
-                         sess.run(predict_op, feed_dict={X: teX, Y: teY,
+                         sess.run(predict_op, feed_dict={X: teX, 
                                                          p_keep_input: 1.0,
                                                          p_keep_hidden: 1.0})))
