Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md
sse_batch.py

README.md

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