转换的tflite模型无法与tflite\u运行时一起使用

2024-03-29 02:33:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个简单的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


Tags: org模型tftensorflow错误pytorchliteexternal