当ONNX模型的opsetversion大于7时，GEMM操作不再包含broadcast参数。然而，onnx2caffe转换工具仍按照旧版方式进行转换，导致转换错误。解决方法是修改_on_operaters.py文件，删除涉及broadcast参数的判断条件，仅保留对transB参数的检查。这样可以确保新版本ONNX模型能正确转换为Caffe格式。
onnx的 opset version> = 7的gemm函数已经去掉了broadcast参数，而onnx2caffe的_operaters.py文件中是按照opset version< 7的方式进行的转换，所以如果你的onnx版本比较新的话就会报如题所示错误，解决方法：
将含有这个参数的判断条件中的相关条件去掉即可，例如：
if node.attrs["broadcast"] != 1 or node.attrs["transB"] != 1:
直接修改为：
if node.attrs["transB"] != 1:
扫一扫
8561
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
