78ba1c8cfcd2c31f1e1b007d2e20e771049e869b,tensorboardX/graph.py,,graph,#Any#Any#,35

Before Change

def graph(model, lastVar):
    global id2name
    global list_of_nodes
    id2name = {id(m): n.replace(".", "/") + "(parameters)" for n, m in model.named_parameters()}
    nodes = []
    list_of_nodes = []

    make_list_of_nodes(lastVar.grad_fn)

After Change

        trace, _ = torch.jit.trace(model, args)
    torch.onnx._optimize_trace(trace, False)
    graph = trace.graph()
    print(graph)
    list_of_nodes = parse(graph)
    nodes = []
    for node in list_of_nodes:
        nodes.append(

Instances