search

MTLab / onnx2caffe

pytorch to caffe by onnx
MIT License
370 stars 102 forks source link

pytorch1.5 onnx2caffe failed #34

Open qiuhaining opened 4 years ago

qiuhaining commented 4 years ago

from pytorch1.5 to onnx,then convert onnx to caffemodel with your code but the output of caffemodel is zero why?

onnx_pytorchv1.5.zip

shopping-tang commented 4 years ago

me too,and i get erro: if node.attrs["broadcast"] != 1 or node.attrs["transB"] != 1: KeyError: 'broadcast'

shopping-tang commented 4 years ago

i tried to change "if node.attrs["broadcast"] != 1 or node.attrs["transB"] != 1:" to "if node.attrs["transB"] != 1:"，It works and i get a caffe model. But when I run the caffe model,i get a new bug:"Exception: Input blob arguments do not match net inputs."

shopping-tang commented 4 years ago

OK，I got it ! I repaired this bug ！

lxgyChen commented 4 years ago

OK，I got it ! I repaired this bug ！

what did yo do...

shopping-tang commented 4 years ago

OK，I got it ! I repaired this bug ！

what did yo do...

Sorry, bro，I notice your email now, my method is what I said above.