pulsar输出模型信息出现错误

使用tf2onnx转pb为onnx模型成功,且使用onnxruntime调用模型成功,但是用pulsar info model.onnx的信息时候出现如下问题,我需要在哪一步进行修改。

img

从错误信息来看,模型在转换过程中丢失了一些信息,导致Pulsar无法完整解析该模型。
可能的原因有:

  1. TensorFlow模型暴露的接口不完整。Pulsar需要模型输出、输入节点信息以解析模型,如果 TensorFlow 暴露的接口不完整,可能导致解析失败。
  2. ONNX转换过程中信息丢失。TensorFlow 到 ONNX 的转换工具还不够健壮,可能在转换过程中丢失一些信息,导致 Pulsar 无法解析。
  3. Pulsar 的 ONNX 支持还不完善。Pulsar 对 ONNX 的支持还在提高中,对某些 ONNX 模型的解析可能还不够健壮。
    建议的解决方案:
  4. 检查 TensorFlow 模型接口,尽量保证关键信息暴露完整。
  5. 尝试不同的转换工具,或不同版本的转换工具。有时不同工具和版本的转换效果会有差异。
  6. 提 issue 给 Pulsar,报告模型转换的相关信息,以帮助 Pulsar 进一步提高 ONNX 支持。
  7. 尝试调整 ONNX 模型,有时很小的调整就可以解决兼容性问题。但这个方法需要对 ONNX 和 Pulsar 有一定了解,建议尝试后面两种方法先。