TensorFlow QAT如何获得量化的权重

Question

如何在TensorFlow的量化感知训练后得到量化权值或量化感知权值。我想在应用程序的NumPy上使用整数量化的权重。到目前为止，我的用法如下，它给出了float32的值。

import tensorflow.keras.backend as K

for layer in q_aware_model.trainable_weights:
    weights_temp.append(K.get_value(layer))

Answer 1

要查看量化后的权重，必须先转换模型。请参阅this TF example中的流程描述和numpy应用程序

请注意，您可能需要使用夜间tf构建才能使量化函数正常工作。