Skip to content
Snippets Groups Projects
Commit 435201ee authored by Yizhi Liu's avatar Yizhi Liu Committed by Tianqi Chen
Browse files

fix mxnet model import (#449)

parent 4b765c51
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment