【发布时间】:2017-04-08 20:24:40
【问题描述】:
网上有教程展示了如何量化.pb TensorFlow 模型,请参阅:
https://petewarden.com/2016/05/03/how-to-quantize-neural-networks-with-tensorflow/
我想知道是否有办法在使用tf.train.write_graph() 保存.pb 文件之前使用python 量化图形
换句话说,有没有像quantize(graph_def) 这样的函数,我可以在保存之前运行它来将图形量化为 8 位权重和操作,从而省去了保存文件后必须通过命令行进行操作的麻烦(就像上面链接的教程大纲一样)。
【问题讨论】:
-
这将非常有用,否则我们必须运行繁重的 bazel build 。
标签: python tensorflow quantization