TensorFlow:保存前使用python量化模型
在线教程展示了如何量化TensorFlow:保存前使用python量化模型,python,tensorflow,quantization,Python,Tensorflow,Quantization,在线教程展示了如何量化.pbTensorFlow模型,请参见: 我想知道的是,在使用tf.train.write_graph() 换句话说,是否有类似于quantize(graph_def)的功能,我可以在保存之前将图形量化为8位权重和操作,省去了保存文件后必须通过命令行执行的麻烦(如上面链接的教程).您可以直接从Python使用图形转换工具的和规则。下面是一个例子: 这将非常有用,因为否则我们必须运行大量bazel构建。
.pb
TensorFlow模型,请参见:
我想知道的是,在使用tf.train.write_graph()
换句话说,是否有类似于quantize(graph_def)
的功能,我可以在保存之前将图形量化为8位权重和操作,省去了保存文件后必须通过命令行执行的麻烦(如上面链接的教程).您可以直接从Python使用图形转换工具的和规则。下面是一个例子:
这将非常有用,因为否则我们必须运行大量bazel构建。