dmlc--dgl
Lingfan Yu
f4a9a455f5
* fix all code tutorial links and typos in texts * sse readme format * fix * sse paper link * gat readme * fix
35 行
1.2 KiB
Markdown
35 行
1.2 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)
|
|
|
|
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 100 \
|
|
--lr 0.001 \
|
|
--batch-size 1024 \
|
|
--dgl \
|
|
--use-spmv \
|
|
--neigh-expand 4
|
|
```
|