hfl

Author: Hsy-Intel

cczoo/horizontal_fl/hfl-tensorflow/__pycache__/cifar10_input.cpython-36.pyc

@@ -50,22 +50,15 @@ function make_custom_env() {
 ROLE=$1
 PS_HOSTS=$2
 WORKER_HOSTS=$3
+
 if [ "$ROLE" == "make" ]; then
-    # rm -rf model *.log
     make clean && make | make_logfilter
 elif [ "$ROLE" == "ps0" ]; then
     make_custom_env
     taskset -c 0-3 stdbuf -o0 gramine-sgx python -u train.py --task_index=0 --job_name=ps $PS_HOSTS $WORKER_HOSTS 2>&1 | runtime_logfilter | tee -a ps0-gramine-python.log &
     if [ "$DEBUG" != "0" ]; then
         wait && kill -9 `pgrep -f gramine`
     fi
-elif [ "$ROLE" == "ps1" ]; then
-    make_custom_env
-    taskset -c 4-7 stdbuf -o0 gramine-sgx python -u train.py --task_index=1 --job_name=ps $PS_HOSTS $WORKER_HOSTS 2>&1  | runtime_logfilter | tee -a ps1-gramine-python.log &
-    if [ "$DEBUG" != "0" ]; then
-        wait && kill -9 `pgrep -f gramine`
-    fi
-
 elif [ "$ROLE" == "worker0" ]; then
     make_custom_env
     taskset -c 8-11 stdbuf -o0 gramine-sgx python -u train.py --task_index=0 --job_name=worker $PS_HOSTS $WORKER_HOSTS 2>&1 | runtime_logfilter | tee -a worker0-gramine-python.log &

cczoo/horizontal_fl/hfl-tensorflow/__pycache__/resnet_model.cpython-36.pyc

@@ -39,9 +39,6 @@ WORKER_HOSTS=$3
 if [ "$ROLE" == "ps0" ]; then
     make_custom_env
     taskset -c 0-3 stdbuf -o0 python -u train.py --task_index=0 --job_name=ps 2>&1 $PS_HOSTS $WORKER_HOSTS | tee -a ps0-python.log &
-elif [ "$ROLE" == "ps1" ]; then
-    make_custom_env
-    taskset -c 4-7 stdbuf -o0 python -u train.py --task_index=1 --job_name=ps 2>&1 $PS_HOSTS $WORKER_HOSTS | tee -a ps1-python.log &
 elif [ "$ROLE" == "worker0" ]; then
     make_custom_env
     taskset -c 8-11 stdbuf -o0 python -u train.py --task_index=0 --job_name=worker $PS_HOSTS $WORKER_HOSTS 2>&1 | tee -a worker0-python.log &