dmlc--dgl
Chao Ma
f8ae6350b1
* update metis * update * update dataloader * update dataloader * new script * update * update * update * update * update * update * update * update dataloader * update * update * update * update * update
26 行
369 B
Markdown
26 行
369 B
Markdown
## Training Scripts for distributed training
|
|
|
|
1. Partition data
|
|
|
|
Partition FB15k:
|
|
|
|
```bash
|
|
./partition.sh ../data FB15k 4
|
|
```
|
|
|
|
Partition Freebase:
|
|
|
|
```bash
|
|
./partition.sh ../data Freebase 4
|
|
```
|
|
|
|
2. Modify `ip_config.txt` and copy dgl-ke to all the machines
|
|
|
|
3. Run
|
|
|
|
```bash
|
|
./launch.sh \
|
|
~/dgl/apps/kg/distributed \
|
|
./fb15k_transe_l2.sh \
|
|
ubuntu ~/mykey.pem
|
|
``` |