onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Load model from ./prefix_tune/robert_large_post_v1_test/model.onnx failed:Type Error: Type parameter (T) of Optype (Concat) bound to different types (tensor(float) and tensor(int64) in node (Concat_7).这个问题的原因是什么,怎么解决?
时间: 2023-06-14 15:04:04 浏览: 1742
tensorflow ImportError: DLL load failed: 找不到指定的模块。
这个错误是由于模型中的 `Concat` 操作绑定了两种不同的类型参数 `tensor(float)` 和 `tensor(int64)`,导致类型不匹配,从而无法加载模型。
为了解决这个问题,你可以尝试以下几个方法:
1. 确认你的模型输入和输出的数据类型是否正确。如果输入的数据类型是 `float`,则确保所有的操作都使用了 `float` 类型的参数。如果输入的数据类型是 `int64`,则确保所有的操作都使用了 `int64` 类型的参数。
2. 确认你的代码是否正确。如果你在自己的代码中使用了 `Concat` 操作,确保你将所有的输入都转换为相同的数据类型。如果你使用了第三方库或框架,确保它们都正确地处理了数据类型。
3. 如果你使用的是 ONNXRuntime 的 Python API,可以尝试升级到最新版本的 ONNXRuntime,看看是否能够解决问题。你也可以尝试使用其他的 ONNXRuntime 接口,例如 C++ API。
4. 如果你无法解决这个问题,你可以考虑重新训练模型,确保所有的操作都使用了相同的数据类型。
阅读全文