Skip to content
This repository has been archived by the owner on Aug 30, 2018. It is now read-only.

Fail to convert caffe2 module to onnx module. #177

Open
Lucasyc opened this issue Feb 3, 2018 · 2 comments
Open

Fail to convert caffe2 module to onnx module. #177

Lucasyc opened this issue Feb 3, 2018 · 2 comments

Comments

@Lucasyc
Copy link

Lucasyc commented Feb 3, 2018

I can't convert a caffe2 module to onnx module.
I first tried the example on README and then write a MINST module myself, both of them generated the following error messages.
I am using x86_64 w/ Ubuntu 14.04 and no GPUs currently.
Thanks in advance for any suggestions!

Traceback (most recent call last):
File "caffe2toonnx.py", line 25, in
value_info,
File "/users/lucasyc/anaconda2/lib/python2.7/site-packages/onnx_caffe2/frontend.py", line 517, in caffe2_net_to_onnx_model
model = make_model(cls.caffe2_net_to_onnx_graph(*args, **kwargs))
File "/users/lucasyc/anaconda2/lib/python2.7/site-packages/onnx_caffe2/frontend.py", line 357, in caffe2_net_to_onnx_graph
cls._ssa_rewrite(predict_net, init_net, value_info)
File "/users/lucasyc/anaconda2/lib/python2.7/site-packages/onnx_caffe2/frontend.py", line 492, in _ssa_rewrite
assert re.match('GivenTensor.*Fill', op.type)
AssertionError

@houseroad
Copy link
Member

This may be caused by some fake initializer of real input in the init_net.pb.
I will create a PR to address this.

@houseroad
Copy link
Member

facebookarchive/caffe2#2058 should be fixed here.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants