fix reddit self loop (#2330)

88a042d3 · Quan (Andy) Gan · GitHub · 4fb0241b · 88a042d3
Unverified Commit 88a042d3 authored Nov 09, 2020 by Quan (Andy) Gan Committed by GitHub Nov 09, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/graphsage/train_sampling_unsupervised.py examples/pytorch/graphsage/train_sampling_unsupervised.py +1 -1

No files found.
--- a/examples/pytorch/graphsage/train_sampling_unsupervised.py
+++ b/examples/pytorch/graphsage/train_sampling_unsupervised.py
@@ -288,7 +288,7 @@ def run(proc_id, n_gpus, args, devices, data):

 def main(args, devices):
    # load reddit data
-    data = RedditDataset(self_loop=True)
+    data = RedditDataset(self_loop=False)
    n_classes = data.num_classes
    g = data[0]
    features = g.ndata['feat']