dmlc--dgl
Giuseppe Futia
6c3dba867a
* Update prerequisites of README * dependencies for pytorch models * dependencies for mxnet models * minor
1.4 KiB
1.4 KiB
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