TensorFlow：在保存之前使用python量化模型

Question

有在線教程展示如何量化.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位權重和操作，省去了在保存文件后必須通過命令行執行此操作的麻煩（如上面鏈接的教程）。

Answer 1

您可以直接從Python使用圖形轉換工具的quantize_weights和quantize_nodes規則。 這是一個例子： https ： //github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/graph_transforms/python/transform_graph_test.py#L76