Commit 435201ee by Yizhi Liu Committed by Tianqi Chen

fix mxnet model import (#449)

parent 4b765c51
......@@ -312,7 +312,7 @@ def _from_mxnet_impl(symbol, graph):
attr = symbol.list_attr()
# op_name = symbol.attr('op_name')
childs = symbol.get_children()
if childs:
if childs is not None:
op_name = symbol.attr('op_name')
childs = [_from_mxnet_impl(childs[i], graph) for i in range(len(childs.list_outputs()))]
childs = [x for y in childs for x in _as_list(y)] # expand group symbol
......
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