Skip to content

GitLab

Unverified Commit 870d8d02 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[doc] fix undefined variable in code snippet (#7107)

parent 8cf5ad84
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
docs/source/guide/minibatch-node.rst
View file @ 870d8d02
...@@ -44,6 +44,7 @@ putting the list of generated MFGs onto GPU. ...@@ -44,6 +44,7 @@ putting the list of generated MFGs onto GPU.
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
dataset = gb.BuiltinDataset("ogbn-arxiv").load() dataset = gb.BuiltinDataset("ogbn-arxiv").load()
g = dataset.graph
train_set = dataset.tasks[0].train_set train_set = dataset.tasks[0].train_set
datapipe = gb.ItemSampler(train_set, batch_size=1024, shuffle=True) datapipe = gb.ItemSampler(train_set, batch_size=1024, shuffle=True)
datapipe = datapipe.sample_neighbor(g, [10, 10]) # 2 layers. datapipe = datapipe.sample_neighbor(g, [10, 10]) # 2 layers.
...@@ -205,6 +206,7 @@ of node types to node IDs. ...@@ -205,6 +206,7 @@ of node types to node IDs.
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
dataset = gb.BuiltinDataset("ogbn-mag").load() dataset = gb.BuiltinDataset("ogbn-mag").load()
g = dataset.graph
train_set = dataset.tasks[0].train_set train_set = dataset.tasks[0].train_set
datapipe = gb.ItemSampler(train_set, batch_size=1024, shuffle=True) datapipe = gb.ItemSampler(train_set, batch_size=1024, shuffle=True)
datapipe = datapipe.sample_neighbor(g, [10, 10]) # 2 layers. datapipe = datapipe.sample_neighbor(g, [10, 10]) # 2 layers.
......
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