冻结量化图

时间:2018-06-24 15:36:41

标签: python tensorflow

我目前正在尝试量化Python中的TensorFlow模型。过去,您可以直接量化冻结的模型,但是此功能似乎已从TensorFlow的最新版本中消失了。即您曾经能够做到这一点:

bazel build tensorflow/contrib/quantization/tools:quantize_graph
bazel-bin/tensorflow/contrib/quantization/tools/quantize_graph \
--input=/tmp/classify_image_graph_def.pb \
--output_node_names="softmax" --output=/tmp/quantized_graph.pb \
--mode=eightbit

现在,没有这样的选择。谁能告诉我在TensorFlow 1.8及更高版本中如何将冻结的.pb图转换为量化的8位定点“等效”?

0 个答案:

没有答案