训练好的模型保存与加载
参考: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.加载模型