博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensorflow 模型保存
阅读量:5011 次
发布时间:2019-06-12

本文共 859 字,大约阅读时间需要 2 分钟。

1、首先 

saver = tf.train.Saver(max_to_keep=1) 新建一个saver,max_to_keep是说只保留最后一轮的训练结果 2、使用save方法保存模型
saver.save(sess,"./model_test/"+"CNN_model_test.ckpt") 然后会在./model_test文件夹下生成这么四个文件:

 

meta文件保存的是图结构,meta文件是pb(protocol buffer)格式文件,包含变量、op、集合等。这个保存了计算图的结构

ckpt文件是二进制文件,保存了所有的weights、biases、gradients等变量。在tensorflow 0.11之前,保存在.ckpt文件中。0.11后,通过两个文件保存,如:

MyModel.data-00000-of-00001

MyModel.index

checkpoint文件,该文件是个文本文件,里面记录了保存的最新的checkpoint文件以及其它checkpoint文件列表。在inference时,可以通过修改这个文件,指定使用哪个model

 

加载模型:

加载图:

new_saver = tf.train.import_meta_graph("./model/CNN_model.ckpt.meta") 加载参数:
new_saver.restore(sess,tf.train.latest_checkpoint('./model')) 接下来就可以使用了 如果是在同一个文件中:
new_saver = tf.train.Saver() new_saver.restore(sess,"./model/"+"CNN_model.ckpt")这样就够了

 

参考链接:

https://blog.csdn.net/huachao1001/article/details/78501928

 

转载于:https://www.cnblogs.com/earendil/p/9264314.html

你可能感兴趣的文章
第二百三十一节,Bootstrap 介绍
查看>>
vi/vim 三种模式的操作
查看>>
JAVA面向对象三大特性总结
查看>>
guid
查看>>
Python中出现“TabError: inconsistent use of tabs and spaces in indentation”问题的解决
查看>>
ajax请求
查看>>
js学习总结----DOM增删改和应用
查看>>
希尔伯特矩阵(Hilbert matrix)
查看>>
(20)sopel算法
查看>>
学习总结 javascript 闭包
查看>>
实验吧一个小坑注入
查看>>
【 D3.js 高级系列 — 8.0 】 打标
查看>>
Mac必备软件推荐
查看>>
Android Gson深入分析
查看>>
display:flow-root
查看>>
判读字符串是否为空的全局宏-分享
查看>>
iOS中Block的基础用法
查看>>
mac 终端 使用ftp命令
查看>>
22-reverseString-Leetcode
查看>>
Centos 开机自动联网
查看>>