TensorFlow模型保存与恢复

Tensorflow store/restore model
TensorFlow模型保存与恢复
TensorFlow模型保存与恢复

  1. TensorFlow模型会保存在后缀为.ckpt的文件中。保存后在save这个文件夹中实际会出现3个文件,因为TensorFlow会将计算图的结构和图上参数取值分开保存。model.ckpt.meta文件保存了TensorFlow计算图的结构,可以理解为神经网络的网络结构model.ckpt文件保存了TensorFlow程序中每一个变量的取值checkpoint文件保存了一个目录下所有的模型文件列表

    作者:磨磨唧唧
    链接:https://www.zhihu.com/question/61946760/answer/377658043

  2. tensorflow新版本不会生成.ckpt文件,你只要将这四个文件放入一个文件夹并命名,测试时直接调用这个文件夹就行了,这就相当于旧版本的.ckpt文件。

  3. Tensorflow训练后的模型可以保存checkpoint文件或pb文件。checkpoint文件是结构与权重分离的四个文件,便于训练;pb文件则是graph_def的序列化文件,类似于caffemodel,便于发布和离线预测。官方提供freeze_grpah.py脚本来将ckpt文件转为pb文件。

    作者:jimlee
    链接:https://www.zhihu.com/question/61946760/answer/461033059