博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensorflow初探-checkpoint转换成pb
阅读量:6158 次
发布时间:2019-06-21

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

hot3.png

reader = tf.train.NewCheckpointReader(ckpt_file)with tf.Session() as sess:    for key in reader.get_variable_to_shape_map():       tf.Variable(reader.get_tensor(key), name=key)    #tf.global_variables_initializer()    with tf.gfile.FastGFile(pb_file, "wb") as f:        f.write(sess.graph_def.SerializeToString())

转载于:https://my.oschina.net/u/1384571/blog/878858

你可能感兴趣的文章
学习笔记之Data Visualization
查看>>
Leetcode 3. Longest Substring Without Repeating Characters
查看>>
416. Partition Equal Subset Sum
查看>>
app内部H5测试点总结
查看>>
[TC13761]Mutalisk
查看>>
while()
查看>>
常用限制input的方法
查看>>
IIS7下使用urlrewriter.dll配置
查看>>
并行程序设计学习心得1——并行计算机存储
查看>>
bulk
查看>>
C++ 迭代器运算
查看>>
【支持iOS11】UITableView左滑删除自定义 - 实现多选项并使用自定义图片
查看>>
【算法笔记】多线程斐波那契数列
查看>>
java8函数式编程实例
查看>>
jqgrid滚动条宽度/列显示不全问题
查看>>
在mac OS10.10下安装 cocoapods遇到的一些问题
查看>>
css技巧
查看>>
Tyvj 1728 普通平衡树
查看>>
javascript性能优化
查看>>
多路归并排序之败者树
查看>>