d165905d0ba24cfba414b8e0c20fa8d7c8ab6a6e,nni/retiarii/codegen/pytorch.py,,graph_to_pytorch_model,#Any#Any#Any#,57

Before Change

    return import_pkgs, _PyTorchModelTemplate.format(
        graph_name=("Graph" if graph_name == "_graph" else graph_name),
        inputs=input_code,
        outputs=", ".join(output_names),
        nodes=linebreak.join(node_codes),
        edges=linebreak.join(edge_codes)
    )

After Change

        input_code = "*_inputs"
    else:
        for name in graph.input_node.operation.io_names:
            assert not name.startswith(graph_name)
        input_code = ", ".join(graph.input_node.operation.io_names)

    edge_codes = []
    sorted_nodes = graph.topo_sort()

Instances