[GitHub] [incubator-mxnet] harshitshrma commented on issue #19597: Unable to load vgg16 using mx.model.load_checkpoint

[GitBox <gi...@apache.org>]

harshitshrma commented on issue #19597:
URL: https://github.com/apache/incubator-mxnet/issues/19597#issuecomment-735991470


   @samskalicky I built from the source and then created a sym-link mxnet to my virtual environment.
   I am getting the error while loading the model from the disk:
   
   Code:
   import mxnet as mx
   (symbol, argParams, auxParams) = mx.model.load_checkpoint("vgg16", 0)
   
   Error:
   Traceback (most recent call last):
     File "<stdin>", line 1, in <module>
     File "/home/harshit/.virtualenvs/dl4cv/lib/python3.8/site-packages/mxnet/model.py", line 262, in load_checkpoint
       symbol = sym.load('%s-symbol.json' % prefix)
     File "/home/harshit/.virtualenvs/dl4cv/lib/python3.8/site-packages/mxnet/symbol/symbol.py", line 2820, in load
       check_call(_LIB.MXSymbolCreateFromFile(c_str(fname), ctypes.byref(handle)))
     File "/home/harshit/.virtualenvs/dl4cv/lib/python3.8/site-packages/mxnet/base.py", line 246, in check_call
       raise get_last_ffi_error()
   mxnet.base.MXNetError: MXNetError: Failed loading Op prob of type SoftmaxOutput: [11:21:10] /home/harshit/mxnet/3rdparty/tvm/nnvm/src/core/op.cc:73: Check failed: op != nullptr: Operator SoftmaxOutput is not registered


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org



---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@mxnet.apache.org
For additional commands, e-mail: issues-help@mxnet.apache.org