zoukankan      html  css  js  c++  java
  • 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
  • 相关阅读:
    lua math 库
    【转】Lua 操作系统库
    lua学习笔记(十三)
    lua学习笔记(十二)
    lua学习笔记(十一)
    lua学习笔记(十)
    lua学习笔记(九)
    lua学习笔记(八)
    lua学习笔记(七)
    lua学习笔记(六)
  • 原文地址:https://www.cnblogs.com/wynlfd/p/14781391.html
Copyright © 2011-2022 走看看