ENCP / CNNdroid

Open Source Library for GPU-Accelerated Execution of Trained Deep Convolutional Neural Networks on Android
MIT License
539 stars 181 forks source link

SavePycaffeModelInMessagePack.py, Index out of range Error #34

Closed cscyangj closed 6 years ago

cscyangj commented 6 years ago

HI, I got following error msg when convert the caffe model.

Traceback (most recent call last): File "SavePycaffeModelInMessagePack.py", line 56, in params = get_params(net) File "SavePycaffeModelInMessagePack.py", line 42, in get_params net_params[key] = [net.params[key][0].data, net.params[key][1].data] IndexError: Index out of range

can u help have a look with model attached.

models.zip

latifisalar commented 6 years ago

Hi, sorry for late response. Unfortunately, this model uses PReLU layer, which is not yet supported in the CNNdroid. I can help if you want to implement it yourself.

Thanks

cscyangj commented 6 years ago

okay, thanks, @latifisalar