项目文件夹

文件

T

文件历史

Minjie Wang dec8b49b5d [Examples] Run all the examples (#623 )

* all pytorch examples

* scan through mxnet examples

* change reddit data

* tweak numerical range for unittest

* fix ci

* fix ci

* fix

* add seed to workaround

2019-06-09 00:26:13 -04:00

README.md

[Examples] Run all the examples (#623 )

2019-06-09 00:26:13 -04: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 \
                                      --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 \
                                      --use-spmv \
                                      --neigh-expand 8 \
                                      --n-hidden 16