CRU-Net icon indicating copy to clipboard operation
CRU-Net copied to clipboard

MXNetError: Failed loading Op conv4_x__1_c3x3-b(1)__conv of type ChannelwiseConvolution:

Open shipengai opened this issue 8 years ago • 1 comments

Thanks for your sharing! When I run mx.model.load_checkpoint('CRU-Net-116-ATx28x14-32x4d',0),There are some error:

~/anaconda3/envs/mxnet3/lib/python3.6/site-packages/mxnet/symbol/symbol.py in load(fname) 2419 raise TypeError('fname need to be string') 2420 handle = SymbolHandle() -> 2421 check_call(_LIB.MXSymbolCreateFromFile(c_str(fname), ctypes.byref(handle))) 2422 return Symbol(handle) 2423

~/anaconda3/envs/mxnet3/lib/python3.6/site-packages/mxnet/base.py in check_call(ret) 143 """ 144 if ret != 0: --> 145 raise MXNetError(py_str(_LIB.MXGetLastError())) 146 147 if sys.version_info[0] < 3:

MXNetError: Failed loading Op conv4_x__1_c3x3-b(1)__conv of type ChannelwiseConvolution: [09:17:56] src/core/op.cc:55: Check failed: op != nullptr Operator ChannelwiseConvolution is not registered

I hope your help,Thanks.

shipengai avatar Oct 16 '17 01:10 shipengai

This repo uses an old version of MXNet with a customized ChannelwiseConvolution layer.

You may want to use this MXNet version to load the pre-trained model.

cypw avatar Oct 19 '17 07:10 cypw