项目文件夹

文件

T

文件历史

Da Zheng bea07b41b3 [Sampler] fix the API of neighbor sampler. (#407 )

* don't return aux_info.

* fix sampler test.

* fix sse.

* fix.

* add comment.

2019-02-28 13:45:21 -08:00

README.md

[Tutorial] Update prerequisites of README (#380 )

2019-02-11 22:58:17 -05:00

sse_batch.py

[Sampler] fix the API of neighbor sampler. (#407 )

2019-02-28 13:45:21 -08:00

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