我有一个简单的pytorch模型,我将其转换为ONNX,并最终转换为tflite。 当我加载模型并使用TF.lite进行推理时,一切都很顺利。 但是,当我尝试使用tflite_运行时加载模型并进行推断时,出现以下错误:
RunTimeError: external/org_tensorflow/tensorflow/lite/kernels/add.cc:385 Type INT64 is unsupported by op Add.Node number 70 (ADD) failed to invoke.
以下是我当前用于TF2.6的转换代码:
converter = tf.lite.TFLiteConverter.from_saved_model(path)
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS]
converter.allow_custom_ops=True # if omitted, conversion fails
tflite_rep = converter.convert()
open('exports/deep_snore.tflite', 'wb').write(tflite_rep)
我查看了很多可能的tf博客,但我不知道问题出在哪里。 我能想到的解决方案是用TF重写模型,重新训练它并将其转换为tflite
目前没有回答
相关问题 更多 >
编程相关推荐