README.md 369 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
## 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
```