Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
e0004f71
Commit
e0004f71
authored
May 29, 2019
by
Da Zheng
Committed by
Minjie Wang
May 29, 2019
Browse files
[Doc] sampler readme update. (#581)
parent
306bb3f9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
3 deletions
+5
-3
examples/mxnet/sampling/README.md
examples/mxnet/sampling/README.md
+5
-3
No files found.
examples/mxnet/sampling/README.md
View file @
e0004f71
...
...
@@ -66,12 +66,14 @@ DGLBACKEND=mxnet python3 train.py --model graphsage_cv --batch-size 1000 --test-
### Run multi-processing training
Run the graph store server that loads the reddit dataset with four workers.
When training a GNN model with multiple processes, there are two steps.
Step 1: run a graph store server separately that loads the reddit dataset with four workers.
```
python3 examples/mxnet/sampling/run_store_server.py --dataset reddit --num-workers 4
```
R
un four workers to train GraphSage on the reddit dataset.
Step 2: r
un four workers to train GraphSage on the reddit dataset.
```
python3 ../incubator-mxnet/tools/launch.py -n 4 -s 1 --launcher local python3 multi_process_train.py --model graphsage_cv --batch-size
10
00 --test-batch-size 5000 --n-epochs 1 --graph-name reddit --num-neighbors 1 --n-hidden 128 --dropout 0.2 --weight-decay 0
python3 ../incubator-mxnet/tools/launch.py -n 4 -s 1 --launcher local python3 multi_process_train.py --model graphsage_cv --batch-size
25
00 --test-batch-size 5000 --n-epochs 1 --graph-name reddit --num-neighbors 1 --n-hidden 128 --dropout 0.2 --weight-decay 0
```
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment