[Example][BugFix] Fix Lightning GraphSAGE examples (dmlc#4046) (#4053)

d1124b7b · 彭齐荣 · GitHub · 3c129ad7 · d1124b7b
Unverified Commit d1124b7b authored May 26, 2022 by 彭齐荣 Committed by GitHub May 26, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/pytorch/graphsage/lightning/node_classification.py examples/pytorch/graphsage/lightning/node_classification.py +1 -0

No files found.
--- a/examples/pytorch/graphsage/lightning/node_classification.py
+++ b/examples/pytorch/graphsage/lightning/node_classification.py
@@ -116,6 +116,7 @@ if __name__ == '__main__':
    dataset = DglNodePropPredDataset('ogbn-products')
    graph, labels = dataset[0]
    graph.ndata['label'] = labels.squeeze()
+    graph.create_formats_()
    split_idx = dataset.get_idx_split()
    train_idx, val_idx, test_idx = split_idx['train'], split_idx['valid'], split_idx['test']
    datamodule = DataModule(graph, train_idx, val_idx, [15, 10, 5], 1024, dataset.num_classes)