cluster_train.sh

#!/bin/bash

#export GLOG_v=30
#export GLOG_logtostderr=1

# start pserver0
python -u cluster_train.py \
    --train_data_dir dist_data/dist_train_data \
    --model_output_dir cluster_model \
    --is_local 0 \
    --is_sparse \
    --role pserver \
    --endpoints 127.0.0.1:6000,127.0.0.1:6001 \
    --current_endpoint 127.0.0.1:6000 \
    --trainers 2 \
    > pserver0.log 2>&1 &

# start pserver1
python -u cluster_train.py \
    --train_data_dir dist_data/dist_train_data \
    --model_output_dir cluster_model \
    --is_local 0 \
    --is_sparse \
    --role pserver \
    --endpoints 127.0.0.1:6000,127.0.0.1:6001 \
    --current_endpoint 127.0.0.1:6001 \
    --trainers 2 \
    > pserver1.log 2>&1 &

# start trainer0
#CUDA_VISIBLE_DEVICES=1 python cluster_train.py \
python -u cluster_train.py \
    --train_data_dir dist_data/dist_train_data \
    --model_output_dir cluster_model \
    --use_gpu 0 \
    --is_local 0 \
    --is_sparse \
    --role trainer \
    --endpoints 127.0.0.1:6000,127.0.0.1:6001 \
    --trainers 2 \
    --trainer_id 0 \
    > trainer0.log 2>&1 &

# start trainer1
#CUDA_VISIBLE_DEVICES=2 python cluster_train.py \
python -u cluster_train.py \
    --train_data_dir dist_data/dist_train_data \
    --model_output_dir cluster_model \
    --use_gpu 0 \
    --is_local 0 \
    --is_sparse \
    --role trainer \
    --endpoints 127.0.0.1:6000,127.0.0.1:6001 \
    --trainers 2 \
    --trainer_id 1 \
    > trainer1.log 2>&1 &

echo "2 pservers and 2 trainers started."