4ddd477f356d51d87763a5b27c8948f007691a50,examples/pytorch/graphsage/train_sampling_multi_gpu.py,,,#,244

Before Change

    labels = th.LongTensor(data.labels)
    n_classes = data.num_labels
    // Construct graph
    g = dgl.graph(data.graph.all_edges())
    g.ndata["features"] = features
    prepare_mp(g)
    // Pack data
    data = train_mask, val_mask, in_feats, labels, n_classes, g

    if n_gpus == 1:
        run(0, n_gpus, args, devices, data)
    else:
        procs = []
        for proc_id in range(n_gpus):
            p = mp.Process(target=thread_wrapped_func(run),
                           args=(proc_id, n_gpus, args, devices, data))
            p.start()
            procs.append(p)

After Change

    g, n_classes = load_reddit()
    // Construct graph
    g = dgl.as_heterograph(g)
    in_feats = g.ndata["features"].shape[1]

    if args.inductive:
        train_g, val_g, test_g = inductive_split(g)
    else:

Instances