Benchmark SSE on multi-GPUs

Paper link: http://proceedings.mlr.press/v80/dai18a/dai18a.pdf

Dependencies

• MXNet nightly build
• requests

pip install mxnet --pre
pip install requests

Use a small embedding

DGLBACKEND=mxnet python3 sse_batch.py --graph-file ../../data/5_5_csr.nd \
                                      --n-epochs 1 \
                                      --lr 0.0005 \
                                      --batch-size 1024 \
                                      --use-spmv \
                                      --dgl \
                                      --num-parallel-subgraphs 32 \
                                      --gpu 1 \
                                      --num-feats 100 \
                                      --n-hidden 100

Test convergence

DGLBACKEND=mxnet python3 sse_batch.py --dataset "pubmed" \
                                      --n-epochs 1000 \
                                      --lr 0.001 \
                                      --batch-size 30 \
                                      --dgl \
                                      --use-spmv \
                                      --neigh-expand 8 \
                                      --n-hidden 16