本文共 262 字,大约阅读时间需要 1 分钟。
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:
转载地址:http://parti.baihongyu.com/