保存
model_path = r'C:\Users\。。。\Desktop\model.ckpt'
init = tf.global_variables_initializer()
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(init)
训练
save_path = saver.save(sess, model_path)
print("Model saved in file: %s" % save_path)
重启Spyder
恢复
model_path = r'C:\Users\。。。\Desktop\model.ckpt'
saver = tf.train.Saver()
graph = tf.get_default_graph()
with tf.Session() as sess:
saver.restore(sess, r'C:\Users\yangxing\Desktop\model.ckpt')
继续训练或测试