- 如果存在连续调用saver.restore(sess, save_path),都应该正确地重置图和会话状态,以避免可能的冲突。因此,在调用saver.restore(sess, save_path)之后,可以使用 tf.reset_default_graph() 和 tf.Session() 来清除之前的图和会话状态。比如:
saver = tf.compat.v1.train.Saver()
sess = tf.compat.v1.Session()
sess.run(tf.compat.v1.global_variables_initializer())
save_path = "./model/model"
saver.restore(sess, save_path)
correct_prediction = tf.argmax(y, 1)
result = sess.run(correct_prediction, feed_dict={x: [img1, img2, img3, img4]})
tf.compat.v1.reset_default_graph()##用于来清除之前的图和会话状态
tf.compat.v1.Session()##来清除之前的图和会话状态
sess.close()
找了好久才找到这个解决方法,希望对大家有用!!!