dmlc--dgl
1b5c02c16c
Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com> Co-authored-by: Xinyu Yao <77922129+yxy235@users.noreply.github.com> Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
Multiple GPU Training
Requirements
pip install torchmetrics==0.11.4
How to run
Node classification
Run with following (available dataset: "ogbn-products", "ogbn-arxiv")
python3 node_classification_sage.py --dataset_name ogbn-products
Results with default arguments
* Test Accuracy of "ogbn-products": ~0.7716
* Test Accuracy of "ogbn-arxiv": ~0.6994