dmlc--dgl
Chao Ma
635dfb4a59
* update metis * update * update dataloader * update dataloader * new script * update * update * update * update * update * update * update * update dataloader * update * update * update * update * update * update * update * Add license to every filer header
32 行
1.5 KiB
Bash
可执行文件
32 行
1.5 KiB
Bash
可执行文件
#!/bin/bash
|
|
|
|
##################################################################################
|
|
# This script runing ComplEx model on Freebase dataset in distributed setting.
|
|
# You can change the hyper-parameter in this file but DO NOT run script manually
|
|
##################################################################################
|
|
machine_id=$1
|
|
server_count=$2
|
|
machine_count=$3
|
|
|
|
# Delete the temp file
|
|
rm *-shape
|
|
|
|
##################################################################################
|
|
# Start kvserver
|
|
##################################################################################
|
|
SERVER_ID_LOW=$((machine_id*server_count))
|
|
SERVER_ID_HIGH=$(((machine_id+1)*server_count))
|
|
|
|
while [ $SERVER_ID_LOW -lt $SERVER_ID_HIGH ]
|
|
do
|
|
MKL_NUM_THREADS=1 OMP_NUM_THREADS=1 DGLBACKEND=pytorch python3 ../kvserver.py --model ComplEx --dataset Freebase \
|
|
--hidden_dim 400 --gamma 143.0 --lr 0.1 --total_client 160 --server_id $SERVER_ID_LOW &
|
|
let SERVER_ID_LOW+=1
|
|
done
|
|
|
|
##################################################################################
|
|
# Start kvclient
|
|
##################################################################################
|
|
MKL_NUM_THREADS=1 OMP_NUM_THREADS=1 DGLBACKEND=pytorch python3 ../kvclient.py --model ComplEx --dataset Freebase \
|
|
--batch_size 1024 --neg_sample_size 256 --hidden_dim 400 --gamma 143.0 --lr 0.1 --max_step 12500 --log_interval 100 \
|
|
--batch_size_eval 1000 --neg_sample_size_eval 1000 --test -adv --total_machine $machine_count --num_thread 1 --num_client 40 |