Tensorflow-使用tflite将修剪的模型转换为较低的量化

Question

我正在尝试使用TFLite量化修剪的张量流模型。修剪和量化工作都是独立进行的，但是当我尝试修剪然后量化时，会出现此错误：ValueError: Input 0 of node sequential/prune_low_magnitude_dense/AssignVariableOp was passed float from sequential/prune_low_magnitude_dense/Mul/ReadVariableOp/resource:0 incompatible with expected resource.，而当我尝试进行量化然后修剪时，会出现以下错误：ValueError: 'prune_low_magnitude' can only prune an object of the following types: tf.keras.models.Sequential, tf.keras functional model, tf.keras.layers.Layer, list of tf.keras.layers.Layer. You passed an object of type: bytes.

我应该怎么做才能使这些操作兼容？该模型的类型为tf.keras.models.Sequential，并且出于测试目的，其深度仅为一层。

Answer 1

如本例所示：

https://github.com/tensorflow/model-optimization/blob/master/tensorflow_model_optimization/python/examples/quantization_with_sparsity/keras/mnist_cnn.py

看来您应该进行 strip_pruning。