dmlc--dgl/examples/multigpu

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