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

时间:2017-04-08 20:24:41

标签: python tensorflow quantization

有在线教程展示如何量化.pb TensorFlow模型,请参阅:

https://petewarden.com/2016/05/03/how-to-quantize-neural-networks-with-tensorflow/

我想知道的是,在使用.pb

保存tf.train.write_graph()文件之前,是否有办法使用python量化图表

换句话说,有一些像quantize(graph_def)这样的函数我可以运行以在保存之前将图形量化为8位权重和操作,从而节省了在保存之后必须通过命令行执行此操作的麻烦文件(如上面链接的教程)。

1 个答案:

答案 0 :(得分:4)