Nnvm.frontend.from_mxnet stuck in infinite loop

jmorrill · October 8, 2019, 9:14pm

Happens with a resnet50 based model here:

dmlc/tvm/blob/90b10b80acfa42d22380b7e657af32fb1aa768b3/nnvm/python/nnvm/frontend/mxnet.py#L463


        dep_cnts[name] = len({c.attr('name') for c in childs})
        for child in childs:
            child_name = child.attr('name')
            if child_name not in deps:
                deps[child_name] = set()
            deps[child_name].add(name)
            if child_name not in symbol_map:
                symbol_map[child_name] = child
                queue.append(child)
order = []
while dep_cnts:
    remove = []
    for name in dep_cnts:
        if dep_cnts[name] == 0:
            order.append(symbol_map[name])
            remove.append(name)
            if name in deps:
                for other in deps[name]:
                    dep_cnts[other] -= 1
    for name in remove:
        del dep_cnts[name]