ShuffleNet_V2_pytorch_caffe
ShuffleNet_V2_pytorch_caffe copied to clipboard
I try to convert model from pytorch to caffe but a error occurred:
Traceback (most recent call last): File "/usr/local/lib/python3.6/dist-packages/caffe/net_spec.py", line 160, in _to_proto _param_names[self.type_name] + '_param'), k, v) KeyError: 'ShuffleChannel'
Have you modified the caffe source code to add a new layer ShuffleChannel?
Yes, that is required.
Yes, that is required.
Thanks for the codes, well, could you please also make the caffe "ShuffleChannel" implementation cpps public?