训练好的模型保存与加载

参考:https://blog.csdn.net/albert201605/article/details/79994331

https://blog.csdn.net/weixin_42001089/article/details/81042885

1.保存模型

saver=tf.train.Saver()

with tf.Session() as sess:
sess.run(init)
。。。。
saver.save(sess,'net/my_net’)

训练结束后,在net文件夹下得到:

其中.meta就是我们训练的神经网络图,.data就是训练好的的权值,checkpoint就是最后一次的结果

2.加载模型

posted @ 2019-07-02 15:17  谁动了我的奶盖  阅读(540)  评论(0编辑  收藏  举报