项目文件夹

文件
Chao Ma f8ae6350b1 [DGL-KE] Add METIS preprocessing pipeline (#1365)
* update metis

* update

* update dataloader

* update dataloader

* new script

* update

* update

* update

* update

* update

* update

* update

* update dataloader

* update

* update

* update

* update

* update
2020-03-17 17:15:04 +08:00

369 B

Training Scripts for distributed training

  1. Partition data

Partition FB15k:

./partition.sh  ../data FB15k 4

Partition Freebase:

./partition.sh  ../data Freebase 4
  1. Modify ip_config.txt and copy dgl-ke to all the machines

  2. Run

./launch.sh \
  ~/dgl/apps/kg/distributed \
  ./fb15k_transe_l2.sh \
  ubuntu ~/mykey.pem