TensorFlow模型保存与恢复
Tensorflow store/restore model
-
TensorFlow模型会保存在后缀为.ckpt的文件中。保存后在save这个文件夹中实际会出现3个文件,因为TensorFlow会将计算图的结构和图上参数取值分开保存。model.ckpt.meta文件保存了TensorFlow计算图的结构,可以理解为神经网络的网络结构model.ckpt文件保存了TensorFlow程序中每一个变量的取值checkpoint文件保存了一个目录下所有的模型文件列表
作者:磨磨唧唧
链接:https://www.zhihu.com/question/61946760/answer/377658043 -
tensorflow新版本不会生成.ckpt文件,你只要将这四个文件放入一个文件夹并命名,测试时直接调用这个文件夹就行了,这就相当于旧版本的.ckpt文件。
-
Tensorflow训练后的模型可以保存checkpoint文件或pb文件。checkpoint文件是结构与权重分离的四个文件,便于训练;pb文件则是graph_def的序列化文件,类似于caffemodel,便于发布和离线预测。官方提供freeze_grpah.py脚本来将ckpt文件转为pb文件。
作者:jimlee
链接:https://www.zhihu.com/question/61946760/answer/461033059