Skip to content

T
tic
Commit 51a25982 by ziheng Committed by Tianqi Chen
Options

[FIX] Fix from_mxnet for multiple outputs symbol (#247)

parent cc7cbbe7
Showing with 2 additions and 0 deletions
nnvm/python/nnvm/frontend/mxnet.py
......@@ -365,4 +365,6 @@ def from_mxnet(symbol, arg_params=None, aux_params=None):
else:
msg = "mxnet.Symbol or gluon.HybridBlock expected, got {}".format(type(symbol))
raise ValueError(msg)
if isinstance(sym, list):
sym = _sym.Group(sym)
return sym, params
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment