tensorflow savedmodel 存储模式转成 checkpoint 存储模式

因工作需要,将savedmodel模型转换成checkpoint,网上资料基本都是ckp->savedmodel,特此记录。

Checkpoints

    • 保存全部的模型参数包括网络的和优化器的
    • 文件保存形式:
      ls ./tf_ckpts

      checkpoint
      ckpt-8.data-00000-of-00002
      ckpt-8.data-00001-of-00002
      ckpt-8.index
    • ckpt-8.meta

SavedModel

    • 不仅保存模型参数,还有模型计算过程
    • 可以直接部署到服务器,不需要依赖原始代码
    • 文件保存形式:
      ls {mobilenet_save_path}

      assets #模型依赖的外部文件,比如vocab
      saved_model.pb #模型的网络结构,可以接受tensor输入,计算完后输出tensor
      variables #模型的参数

      ls {mobilenet_save_path}/variables

      variables.data-00000-of-00002
      variables.data-00001-of-00002
      variables.index

SaveModel转成Ckeckpoint

  • 只需要将variables里的文件,改成ckpt.*的形式即可
  • 然后找一个checkpoint的文件,里面放的是ckpt. 文件的地址。把里面地址改下,把文件复制到 variables文件夹下就行了,.pb文件不需要了。
  • 模型指定加载这个variables
posted @ 2021-05-18 16:02  Sunshine168  阅读(651)  评论(0编辑  收藏  举报