dmlc--dgl
Giuseppe Futia
6c3dba867a
* Update prerequisites of README * dependencies for pytorch models * dependencies for mxnet models * minor
46 行
1.4 KiB
Markdown
46 行
1.4 KiB
Markdown
Benchmark SSE on multi-GPUs
|
|
=======================
|
|
|
|
Paper link:
|
|
[http://proceedings.mlr.press/v80/dai18a/dai18a.pdf](http://proceedings.mlr.press/v80/dai18a/dai18a.pdf)
|
|
|
|
Dependencies
|
|
-------------
|
|
* MXNet nightly build
|
|
* requests
|
|
|
|
```bash
|
|
pip install mxnet --pre
|
|
pip install requests
|
|
```
|
|
|
|
Use a small embedding
|
|
---------------------
|
|
|
|
```bash
|
|
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
|
|
----------------
|
|
|
|
```bash
|
|
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
|
|
```
|