Skip to content

GitLab

Unverified Commit 88a042d3 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix reddit self loop (#2330)

parent 4fb0241b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/graphsage/train_sampling_unsupervised.py
View file @ 88a042d3
...@@ -288,7 +288,7 @@ def run(proc_id, n_gpus, args, devices, data): ...@@ -288,7 +288,7 @@ def run(proc_id, n_gpus, args, devices, data):
def main(args, devices): def main(args, devices):
# load reddit data # load reddit data
data = RedditDataset(self_loop=True) data = RedditDataset(self_loop=False)
n_classes = data.num_classes n_classes = data.num_classes
g = data[0] g = data[0]
features = g.ndata['feat'] features = g.ndata['feat']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment