如何保存和使用训练好的模型参数
引言
最近在学习Tensorflow 构建CNN，训练需要耗费时间，把训练好的各个参数保存下来是最简便的，网上有很多教程，但是跟着教程走不一定一帆风顺还是踩了一些坑，然后自己填了一下坑
如何保存好训练的结果：
假设会话为sess，计算图为graph
网上看了很多资料，使用
saver=tf.train.Saver() # 不传入参数代表默认存入全部参数
file_name = 'saved_model/model.ckpt' # 将保存到当前目录下的的saved_model文件夹下model.ckpt文件
saver.saver(sess,file_name ) # 保存好的模型文件
这样来保存模型，这样就行了吗？还不行
但是按照这个方式，开始IDE总是会报错：No Variable to save
然后我的想法是：可能要把sess传给saver是吗？下面是我的代码，和一次尝试性的修改
graph = tf.Graph() # 计算图
with graph.as_default():
# 定义计算图
...
# 以上是一些权重和卷积层的定义，这里就不贴出来了
sess = tf.Session(graph=graph) # 把上一步定义的计算图载入到会话中
# 给定义saver一个sess作为输入，结果也是不行的
saver=tf.train.Saver(sess) # 不传入参数代表默认存入全部参数
saver.saver(sess,'saved_model/model.ckpt')
这样还是会报错：没有可以被用来保存的变量。我思来想去，可能要载入graph?
graph = tf.Graph()
with graph.as_default():
# 定义计算图
...
# ---
sess = tf.Session(graph=graph) # 把上一步定义的计算图载入到会话中
# 这次把graph传进来做参数
saver = tf.train.Saver(graph)
saver.saver(sess,'saved_model/model.ckpt')
结果还是不行
多次尝试之后，终于修改正确了
graph = tf.Graph()
with graph.as_default():
# 定义计算图
...
# ---
saver = tf.train.Saver() # 默认存储上面全部定义参数，如果不想全部存下来，也可以输入你想要保存的参数
sess = tf.Session(graph=graph) # 把上一步定义的计算图载入到会话中
# 保存，这次就可以成功了
saver.saver(sess,'saved_model/model.ckpt')
保存成功，文件夹saved_model下会出现几个文件
checkpoint文件（这个文件很重要，记录了） 还有几个文件，它们的后缀分别是 .data , .index , .meta 。我们似乎可以不搭理这三个文件
先小结一下
如果定义了子图，或者说自己定义graph而不使用tensorflow的默认计算图的时候，定义要在定义graph最后进行定义，想保存哪个子图的变量，就要在哪个子图定义相关的Saver，这样才能实现想要的效果。
Tensorflow的图graph和会话session还是有点抽象的，一不小心就整乱了。
Notes: 假设没有使用with graph.as_graph():这种结构，直接在脚本上定义了tensorflow的变量，再使用saver()应该是没有问题的。
读取和使用保存好的模型参数
那存好了之后，该怎么调用了
假设我需要在另一个新的脚本，例如 ：test.py文件上使用我的代码做测试，要怎么使用保存好的模型参数呢
有两种方式：
先把之前训练，构建计算图已经你定义网络参数的那些代码粘贴到test.py文件下
graph = tf.Graph()
with graph.as_default():
# 定义计算图
...
# ---
saver = tf.train.Saver()
特别注意，运行到上一步的时候
然后再使用如下代码
with tf.Session(graph=graph) as sess:
check_point_path = 'saved_model/' # 保存好模型的文件路径
ckpt = tf.train.get_checkpoint_state(checkpoint_dir=check_point_path)
# 从模型中恢复参数 saver.restore(sess,ckpt.model_checkpoint_path) # 讀取成功，然后就可以使用模型参数进行预测，或者测试了。
如果你觉得上面那个方式有点繁琐，可以直接import train.py
假设train.py这个脚本的代码是这样的：
graph = tf.Graph()
with graph.as_default():
# 定义计算图
...
# ---
saver = tf.train.Saver()
那么你在你的test.py中可以这样写
import train
# 参考python的命名空间方法
graph = train.graph
sess = train.sess
with tf.Session(graph=graph) as sess:
check_point_path = 'saved_model/' # 保存好模型的文件路径
ckpt = tf.train.get_checkpoint_state(checkpoint_dir=check_point_path)
# 从模型中恢复参数 saver.restore(sess,ckpt.model_checkpoint_path)
Notes: 在使用saver.restore(sess,ckpt.model_checkpoint_path) 后，这个时候，就不需要再次使用sess.run(init) 对参数进行初始化了（否则会覆盖掉训练好的参数），如果你在前面使用run，进行初始化，权重会根据你的定义进行初始化，但是你使用这个语句后，模型中的参数会把它覆盖掉
最后再说一下
好像高级的使用方法，可以根据选择不同迭代次数更新时候的权重，这里只做简单总结一下，以后学习到了再更新吧