tensorflow模型的保存与加载

模型保存

#coding:utf-8
import tensorflow as tf
import os

v1=tf.Variable(tf.constant(1.0,shape=[1]),name='v1')
v2=tf.Variable(tf.constant(2.0,shape=[1]),name='v2')
result=v1+v2
init_op=tf.initialize_all_variables()
saver=tf.train.Saver()
with tf.Session() as sess:
    sess.run(init_op)
    saver.save(sess, os.path.join(os.getcwd(), 'model.ckpt'))

　　

加载模型

#coding:utf-8
import tensorflow as tf
import os

v1=tf.Variable(tf.constant(1.0,shape=[1]),name='v1')
v2=tf.Variable(tf.constant(2.0,shape=[1]),name='v2')
result=v1+v2
init_op=tf.initialize_all_variables()
saver=tf.train.Saver()
with tf.Session() as sess:
    saver.restore(sess,'model.ckpt')
    print(sess.run(result))

 

 

 

转载于:https://www.cnblogs.com/yinghuali/p/7430061.html