时间戳

首页 > 解决方案 > 量化转换器错误 2471 检查失败：status.ok()

python - 量化转换器错误 2471 检查失败：status.ok()

问题描述

环境：tensorflow=2.0，tensorflow-model-optimization=0.3.0，python=3.6.8

通过以下代码将 keras 模型转换为 tflite 时：

m1='ownmodel_pruW.h5'
model=tf.keras.models.load_model(m1)
tflite_model_file = 'ownnet.tflite'
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
with open(tflite_model_file, 'wb') as f:
    f.write(tflite_model)

错误如下

ConverterError：有关信息，请参阅控制台。2020-05-14 20:43:12.220536: F tensorflow/lite/toco/import_tensorflow.cc:2471] 检查失败：status.ok() input_content (0) 和 float_val (73723) 都没有正确的尺寸 (73728)这个浮点张量（在处理节点“Modelnet/conv2/Conv2D/ReadVariableOp”时）

有什么方法可以解决这个错误？

标签： pythontensorflowquantization

解决方案

推荐阅读