Hi!
I want to launch one of the mxnet models from here:
Pretrained models are given in archive. I extracted these files to folder:
model-symbol.json
model-0000.params
I installed mxnet with pip3 install mxnet-cu110
Now I try to do this in code:
import mxnet as mx
prefix=’./insight/model’
epoch = ‘0000’
ctx = mx.gpu(0)
sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, epoch)
But it says:
MXNetError Traceback (most recent call last)
in
5
6 ctx = mx.gpu(0)
----> 7 sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, epoch)
8 all_layers = sym.get_internals()
9 sym = all_layers[‘fc1_output’]
~/.local/lib/python3.6/site-packages/mxnet/model.py in load_checkpoint(prefix, epoch)
474 - Parameters will be loaded from prefix-epoch.params
.
475 “”"
→ 476 symbol = sym.load(’%s-symbol.json’ % prefix)
477 arg_params, aux_params = load_params(prefix, epoch)
478 return (symbol, arg_params, aux_params)
~/.local/lib/python3.6/site-packages/mxnet/symbol/symbol.py in load(fname)
3052 raise TypeError(‘fname need to be string’)
3053 handle = SymbolHandle()
→ 3054 check_call(_LIB.MXSymbolCreateFromFile(c_str(fname), ctypes.byref(handle)))
3055 return Symbol(handle)
3056
~/.local/lib/python3.6/site-packages/mxnet/base.py in check_call(ret)
244 “”"
245 if ret != 0:
→ 246 raise get_last_ffi_error()
247
248
MXNetError: std: :bad_alloc
I’m trying to repeat loading process form here:
What can I do with that? Thanks in advance.