d165905d0ba24cfba414b8e0c20fa8d7c8ab6a6e,nni/retiarii/converter/graph_gen.py,,convert_module,#Any#Any#Any#Any#,285

Before Change

    global modules_arg

    assert id(module) in modules_arg, "id not exist: {}, {}".format(id(module), module_name)
    if isinstance(modules_arg[id(module)], tuple):
        positional_args, keyword_args = modules_arg[id(module)]
        m_attrs = keyword_args
        // TODO: remove positional args
        m_attrs["positional_args"] = positional_args
    else:
        m_attrs = modules_arg[id(module)]

    original_type_name = script_module.original_name
    if original_type_name in torch.nn.__dict__ and original_type_name not in MODULE_EXCEPT_LIST:
        // this is a basic module from pytorch, no need to parse its graph
        return None, m_attrs

After Change

    ir_graph._register()

    return ir_graph, modules_arg[id(module)]

def convert_to_graph(script_module, module, recorded_modules_arg):
    

Instances