Merge pull request #58 from gm06041/patch-7

Update 10_save_restore_net.py
nlintz · Oct 18, 2016 · a9f9a5b · a9f9a5b
2 parents 318ee0e + fca3f80
commit a9f9a5b
Showing 1 changed file with 1 addition and 1 deletion.
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})))