发新帖

【机器学习初步】继续做一些优化等操作

[复制链接]
570 1

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
在前面两节中,我们学习了如何开始和训练自己的模型,下面我们继续优化它们。
1. 得到pb类型的图,帮助我们分析
from tensorflow.python.framework import graph_io
graph_io.write_graph(self.sess.graph,"./tests/", "input_graph.pb")


2. 固化freez_graph
bazel build tensorflow/python/tools:freeze_graph&& \
bazel-bin/tensorflow/python/tools/freeze_graph \
--input_graph=some_graph_def.pb \
--input_checkpoint=model.ckpt-8361242 \
--output_graph=/tmp/frozen_graph.pb
--output_node_names=softmax


3. 优化optimize_for_inference
bazel-bin/tensorflow/python/tools/optimize_for_inference
--input=froze_ctc.pb
--output=optimized_ctc.pb
--frozen_graph=True
--input_names=Placeholder,seq_len
--output_names=CTCGreedyDecoder
--placeholder_type_enum=13


4. 量化quantize_graph
bazel-bin/tensorflow/tools/quantization/quantize_graph
--input=./optimized_ctc.pb
--output_node_names=CTCGreedyDecoder
--print_nodes
--output=./quantized_graph.pb
--mode=eightbit
--logtostderr





本楼点评(0) 收起

精彩评论1

neverchange  TF豆豆  发表于 2018-7-4 23:50:24 | 显示全部楼层
这里只列出常用的命令,可以尝试进行操作,遇到什么问题就在这里回帖吧
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表