search

cypw / MXNet2Caffe

Convert MXNet model to Caffe model
MIT License
162 stars 121 forks source link

mxnet #9

Open sunrongda opened 5 years ago

sunrongda commented 5 years ago

Traceback (most recent call last): File "json2prototxt.py", line 38, in write_node(prototxt_file, info) File "/root/MXNet2Caffe/prototxt_basic.py", line 143, in write_node Convolution(txt_file, info) File "/root/MXNet2Caffe/prototxt_basic.py", line 16, in Convolution if info['param']['no_bias'] == 'True': KeyError: 'param'

Tez01 commented 5 years ago

In prototxt_basic.py I think the key should be params. When I print info.keys() output is : [u'inputs', u'name', 'bottom', 'top', 'params', u'attrs', u'op']