因工作需要,将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)
评论()
编辑
收藏
举报