From fca3f8067776f769299dd6c0358f365fd78f7657 Mon Sep 17 00:00:00 2001 From: "Oh, Geun-moon" Date: Tue, 18 Oct 2016 14:48:27 +0900 Subject: [PATCH] Update 10_save_restore_net.py i get mistake. so i edited original source. (Y variable is useless) --- 10_save_restore_net.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/10_save_restore_net.py b/10_save_restore_net.py index 2ca877e..9e92187 100755 --- 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})))