python3 tensorflow 模型的保存与恢复

保存

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')

继续训练或测试