MXNetError: Failed loading Op conv4_x__1_c3x3-b(1)__conv of type ChannelwiseConvolution:
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.
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.