02d3197407487646dc5ca6abd889b8fe5fed1aef,examples/pytorch/graphsage/train_sampling.py,,,#,223

Before Change

        device = th.device("cpu")

    // load reddit data
    data = RedditDataset(self_loop=True)
    train_mask = data.train_mask
    val_mask = data.val_mask
    features = th.Tensor(data.features)
    in_feats = features.shape[1]
    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

After Change

    if args.dataset == "reddit":
        g, n_classes = load_reddit()
    elif args.dataset == "ogb-product":
        g, n_classes = load_ogb("ogbn-products")
    else:
        raise Exception("unknown dataset")
    g = dgl.as_heterograph(g)

Instances