tensorflow模型保存

首先调用方法:

saver = tf.train.Saver() 

保存模型:

saver.save(sess, "model/model.ckpt")  

注意,要写上文件路径,直接写文件名会报错

 

加载路径:

saver.restore(sess, "model/model.ckpt-24000")

这里要加上训练的轮数,在你的model里会出现,这个问题困扰了很多人,最终解决

posted on 2018-01-11 10:50  特立独行的十楼  阅读(106)  评论(0)    收藏  举报

导航